mindspore.Tensor.item

mindspore.Tensor.item(index=None)[源代码]

获取Tensor中指定索引的元素。

说明

Tensor.item返回的是Tensor标量,而不是Python标量。

参数:
  • index (Union[None, int, tuple(int)]) - Tensor的索引。默认值:None。

返回:

Tensor标量,dtype与原始Tensor的相同。

异常:
  • ValueError - index 的长度不等于Tensor的ndim。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1,2,3],[4,5,6]], dtype=np.float32))
>>> x = x.item((0,1))
>>> print(x)
2.0