mindspore.mint.narrow
- mindspore.mint.narrow(input, dim, start, length)[源代码]
沿着指定的轴,指定起始位置获取指定长度的Tensor。
- 参数:
input (Tensor) - 需要计算的Tensor。
dim (int) - 指定的轴。
start (int) - 指定起始位置。
length (int) - 指定长度。
- 返回:
Tensor。
- 异常:
ValueError - dim 值超出范围[-input.ndim, input.ndim)。
ValueError - start 值超出范围[-input.shape[dim], input.shape[dim]]。
ValueError - length 值超出范围[0, input.shape[dim]-start]。
- 支持平台:
Ascend
样例:
>>> import mindspore >>> from mindspore import mint >>> from mindspore import Tensor >>> x = Tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], mindspore.int32) >>> output = mint.narrow(x, 0, 0, 2) >>> print(output) [[ 1 2 3] [ 4 5 6]] >>> output = mint.narrow(x, 1, 1, 2) >>> print(output) [[ 2 3] [ 5 6] [ 8 9]]