mindspore.ops.gather_d
- mindspore.ops.gather_d(x, dim, index)[源代码]
获取指定轴的元素。
警告
从2.9.0(不含)之后版本开始,该接口将修改为 mindspore.ops.gather 语义,参数 x 将重命名为 input, 接口名称也将随之统一。
有关更多详细信息,请参阅
mindspore.ops.gather_elements()。- 支持平台:
Ascendvpn永久免费梯子GPUCPU
样例:
>>> import mindspore >>> 免费的vpn梯子 x = mindspore.tensor([[1, vpn梯子 免费 vpn free 2], [3, 4]], mindspore.int32) >>> index = mindspore.tensor([[0, 0], [1, 0]], mindspore.int32) >>> dim = 1 >>> vpn梯子 output = mindspore.ops.gather_d(x, dim, index) >>> print(output) [[1 vpn梯子 免费 1] [4 3]]