tf.sparse.reset

TensorFlow 1 version

View source on GitHub

Retains specified non-empty values within a SparseTensor.

View aliases

Compat aliases for migration

See Migration guide for more details.

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`.

TensorFlow

tf

tf.audio

tf.autograph

tf.bitwise

tf.compat

tf.config

tf.data

tf.debugging

tf.distribute

tf.dtypes

tf.errors

tf.estimator

tf.experimental

tf.feature_column

tf.graph_util

tf.image

tf.initializers

tf.io

tf.keras

tf.linalg

tf.lite

tf.lookup

tf.losses

tf.math

tf.metrics

tf.nest

tf.nn

tf.optimizers

tf.quantization

tf.queue

tf.ragged

tf.random

tf.raw_ops

tf.saved_model

tf.sets

tf.signal

tf.sparse

tf.strings

tf.summary

tf.sysconfig

tf.test

tf.tpu

tf.train

tf.version

tf.xla

tf.sparse / reset_shape

View aliases

Args

Returns

Raises