TensorFlow 1 version | View source on GitHub |
Masks elements of IndexedSlices
.
tf.sparse.mask(
a, mask_indices, name=None
)
Given an IndexedSlices
instance a
, returns another IndexedSlices
that
contains a subset of the slices of a
. Only the slices at indices not
specified in mask_indices
are returned.
This is useful when you need to extract a subset of slices in an
IndexedSlices
object.
For example:
# `a` contains slices at indices [12, 26, 37, 45] from a large tensor
# with shape [1000, 10]
a.indices # [12, 26, 37, 45]
tf.shape(a.values) # [4, 10]
# `b` will be the subset of `a` slices at its second and third indices, so
# we want to mask its first and last indices (which are at absolute
# indices 12, 45)
b = tf.sparse.mask(a, [12, 45])
b.indices # [26, 37]
tf.shape(b.values) # [2, 10]
Args | |
---|---|
a
|
An IndexedSlices instance.
|
mask_indices
|
Indices of elements to mask. |
name
|
A name for the operation (optional). |
Returns | |
---|---|
The masked IndexedSlices instance.
|