tf.sparse.retain

Retains specified non-empty values within a SparseTensor.

Compat aliases for migration

tf.compat.v1.sparse.retain, tf.compat.v1.sparse_retain

tf.sparse.retain(
    sp_input, to_retain
)

For example, if sp_input has shape [4, 5] and 4 non-empty string values:

[0, 1]: a
[0, 3]: b
[2, 0]: c
[3, 1]: d

and to_retain = [True, False, False, True], then the output will be a SparseTensor of shape [4, 5] with 2 non-empty values:

[0, 1]: a
[3, 1]: d

Args
`sp_input`	The input `SparseTensor` with `N` non-empty elements.
`to_retain`	A bool vector of length `N` with `M` true values.

Returns
A `SparseTensor` with the same shape as the input and `M` non-empty elements corresponding to the true positions in `to_retain`.

Raises
`TypeError`	If `sp_input` is not a `SparseTensor`.

© 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.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/sparse/retain

Title here