mindspore.Tensor.item
- mindspore.Tensor.item(index=None)[源代码]
获取Tensor中指定索引的元素。
- 参数:
index (Union[None, int, tuple(int)]) - Tensor的索引。默认值:
None
。
- 返回:
标量,类型由Tensor的dtype决定。
- 异常:
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