mindspore.ops.narrow
- mindspore.ops.narrow(input, axis, start, length)[源代码]
沿着指定的轴,指定起始位置获取指定长度的Tensor。
- 参数:
input (Tensor) - 需要计算的Tensor。
axis (int) - 指定的轴。
start (int) - 指定起始位置。
length (int) - 指定长度。
- 返回:
Tensor。
output (Tensors) - 计算后的Tensor。
- 异常:
TypeError - 如果输入不是Tensor、元组或Tensor组成的列表。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> from mindspore import ops >>> from mindspore import Tensor >>> x = Tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], mindspore.int32) >>> output = ops.narrow(x, 0, 0, 2) >>> print(output) [[ 1 2 3] [ 4 5 6]] >>> output = ops.narrow(x, 1, 1, 2) >>> print(output) [[ 2 3] [ 5 6] [ 8 9]]