mindspore.Tensor.narrow

mindspore.Tensor.narrow(dim, start, length)

沿着指定的轴，指定起始位置获取指定长度的Tensor。

参数：

• dim (int) - 指定的轴。

• start (Union[int, Tensor]) - 指定起始位置。

• length (int) - 指定长度。

返回：

output (Tensor) - narrow后的Tensor。

异常：

• ValueError - dim 值超出范围[-self.ndim, self.ndim)。

• ValueError - start 值超出范围[-self.shape[dim], self.shape[dim]]。

• ValueError - length 值超出范围[0, self.shape[dim]-start]。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> from mindspore import Tensor
>>> x = Tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], mindspore.int32)
>>> output = x.narrow(0, 0, 2)
>>> print(output)
[[ 1 2 3]
 [ 4 5 6]]
>>> output = x.narrow(1, 1, 2)
>>> print(output)
[[ 2 3]
 [ 5 6]
 [ 8 9]]