mindspore.Tensor.item

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

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

说明

Tensor.item返回的是Tensor标量,而不是Python标量。如果这个Tensor是一个Tensor标量,Tensor.item返回的是numpy.ndarray。

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

返回:

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

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

支持平台:

Ascend GPU CPU

样例:

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