mindspore.ops.slice

mindspore.ops.slice(input_x, begin, size)

根据指定shape对输入tensor进行切片。

说明

begin 的起始值为0，size 的起始值为1。

如果 size[i] 为-1，则维度i中的所有剩余元素都包含在切片中。这相当于 $size[i]=input\mathrm{\_}x.shape(i)-begin[i]$ 。

参数：

• input_x (Tensor) - 输入tensor。

• begin (Union[tuple, list]) - 切片的起始位置，表示每个维度的偏移。

• size (Union[tuple, list]) - 切片的大小。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> data = mindspore.tensor([[[1, 1, 1], [2, 2, 2]],
...                         [[3, 3, 3], [4, 4, 4]],
...                         [[5, 5, 5], [6, 6, 6]]], mindspore.int32)
>>> output = mindspore.ops.slice(data, (1, 0, 0), (1, 1, 3))
>>> print(output)
[[[3 3 3]]]
>>> output = mindspore.ops.slice(data, (1, 0, 0), (1, 1, 2))
>>> print(output)
[[[3 3]]]
>>> output = mindspore.ops.slice(data, (1, 0, 0), (1, 1, 1))
>>> print(output)
[[[3]]]
>>> output = mindspore.ops.slice(data, (1, 1, 0), (1, 1, 3))
>>> print(output)
[[[4 4 4]]]
>>> output = mindspore.ops.slice(data, (1, 0, 1), (1, 1, 2))
>>> print(output)
[[[3 3]]]