mindspore.Tensor.narrow

mindspore.Tensor.narrow(axis, start, length)[源代码]

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

参数:
  • axis (int) - 指定的轴。

  • start (int) - 指定的起始位置。

  • length (int) - 指定的长度。

返回:

Tensor。

异常:
  • TypeError - axis不是int类型。

  • TypeError - start不是int类型。

  • TypeError - length不是int类型。

  • ValueError - axis取值不在[0, ndim-1]范围内。

  • ValueError - start取值不在[0, shape[axis]-1]范围内。

  • ValueError - start+length超出Tensor的维度范围shape[axis]-1。

支持平台:

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]]