mindspore.Tensor.type
- mindspore.Tensor.type(dtype=None)[源代码]
将Tensor的dtype转换成 dtype 。如果 dtype 为
None
,则返回原类型。- 参数:
dtype (mindspore.dtype,可选) - 指定输出Tensor的数据类型。默认值:
None
。
- 返回:
Tensor或str。如果 dtype 为
None
则返回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]]