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