experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add an example of an IndexedSlices

Browse Source 
PiperOrigin-RevId: 346108545
Change-Id: I9ba894e07e2f2d0c3884c1834a5a2c2308451a91
This commit is contained in:
Mark Daoust 2020-12-07 09:21:49 -08:00 committed by TensorFlower Gardener
parent 03d7d7667f
commit 4479e94258
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/python/framework/indexed_slices.py
View File

@ -81,6 +81,18 @@ class IndexedSlices(internal.NativeObject, composite_tensor.CompositeTensor):
gradients for operations that have sparse gradients
(e.g. `tf.gather`).
>>> v = tf.Variable([[0.,1, 2], [2, 3, 4], [4, 5, 6], [6, 7, 8]])
>>> with tf.GradientTape() as tape:
... r = tf.gather(v, [1,3])
>>> index_slices = tape.gradient(r,v)
>>> index_slices
<...IndexedSlices object ...>
>>> index_slices.indices.numpy()
array([1, 3], dtype=int32)
>>> index_slices.values.numpy()
array([[1., 1., 1.],
[1., 1., 1.]], dtype=float32)
Contrast this representation with
`tf.sparse.SparseTensor`,
which uses multi-dimensional indices and scalar values.