mindspore.Tensor.type

查看源文件
mindspore.Tensor.type(dtype=None)[源代码]

将Tensor的dtype转换成 dtype 。如果 dtypeNone ,则返回原类型。

参数:
  • dtype (mindspore.dtype,可选) - 指定输出Tensor的数据类型。默认值: None

返回:

Tensor或str。如果 dtypeNone 则返回str,str描述了Tensor的数据类型。如果 dtype 不为 None ,则返回Tensor,返回Tensor的dtype是 dtype

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> from mindspore import Tensor
>>> x = Tensor([[1.2, 2], [3.4, 4]], dtype=mindspore.float32)
>>> print(x.type())
Float32
>>> print(x.type(dtype=mindspore.int32))
[[1 2]
 [3 4]]