tf.sparse.slice  |  TensorFlow v2.16.1 (original) (raw)

tf.sparse.slice

Stay organized with collections Save and categorize content based on your preferences.

Slice a SparseTensor based on the start and size.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.sparse.slice, tf.compat.v1.sparse_slice

tf.sparse.slice(
    sp_input, start, size, name=None
)

Used in the notebooks

Used in the guide
Working with sparse tensors

For example, if the input is

input_tensor = shape = [2, 7]
[    a   d e  ]
[b c          ]

Graphically the output tensors are:

sparse.slice([0, 0], [2, 4]) = shape = [2, 4]
[    a  ]
[b c    ]

sparse.slice([0, 4], [2, 3]) = shape = [2, 3]
[ d e  ]
[      ]
Args
sp_input The SparseTensor to split.
start 1-D. tensor represents the start of the slice.
size 1-D. tensor represents the size of the slice.
name A name for the operation (optional).
Returns
A SparseTensor objects resulting from splicing.
Raises
TypeError If sp_input is not a SparseTensor.