tf.sparse.slice
| View source on GitHub |
Slice a SparseTensor based on the start and `size.
tf.sparse.slice(
sp_input, start, size, name=None
)
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. |
© 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.4/api_docs/python/tf/sparse/slice