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