oneflow.narrow

oneflow.narrow(x, dim: int, start: int, length: int) → Tensor

Returns a new tensor that is a narrowed version of input tensor. The dimension dim is input from start to start + length.

Parameters

• input – the tensor to narrow.

• dim – the dimension along which to narrow.

• start – the starting dimension.

• length – the distance to the ending dimension.

For example:

>>> import oneflow as flow
>>> input = flow.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> flow.narrow(input, 0, 0, 2)
tensor([[1, 2, 3],
        [4, 5, 6]], dtype=oneflow.int64)
>>> flow.narrow(input, 1, 1, 2)
tensor([[2, 3],
        [5, 6],
        [8, 9]], dtype=oneflow.int64)