mindspore.mint.select

mindspore.mint.select(input, dim, index)

在给定索引处沿选定维度对输入张量进行切片。

警告

这是一个实验性API，可能会更改或删除。

参数：

• input (Tensor) - 输入Tensor。

• dim (int) - 进行切片的维度。

• index (int) - 要选择的索引。

返回：

Tensor。

异常：

• TypeError - 如果 input 不是Tensor。

支持平台：

Ascend

样例：

>>> import mindspore
>>> from mindspore import Tensor, mint
>>> input = Tensor([[2, 3, 4, 5],[3, 2, 4, 5]])
>>> y = mint.select(input, 0, 0)
>>> y = Tensor([1,2], mindspore.float32)
>>> print(y)
[2 3 4 5]