mindspore.Tensor.item
- mindspore.Tensor.item(index=None)[源代码]
获取Tensor中指定索引的元素。
Note
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