|View source on GitHub|
Masks elements of
`tf.sparse_mask`Compat aliases for migration
See Migration guide for more details.
tf.sparse.mask( a, mask_indices, name=None )
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
# `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]
| || An |
| ||Indices of elements to mask.|
| ||A name for the operation (optional).|
| The masked |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.