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