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