tf_agents.utils.composite.slice_from

Slice a composite tensor along axis from start.

tf_agents.utils.composite.slice_from(
    tensor, axis, start
)

Examples:

slice_from(tensor, 2, 1) === tensor[:, :, 1:]
sparse_to_dense(slice_from(sparse_tensor, 2, 1))
  === sparse_to_dense(sparse_tensor)[:, :, 1:]

Args
tensor	A Tensor or SparseTensor.
axis	A python integer.
start	A 0D scalar.

Returns
The sliced composite tensor.