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.