mindspore.dataset.transforms.TypeCast

class mindspore.dataset.transforms.TypeCast(data_type)[源代码]

将输入的Tensor转换为指定的数据类型。

说明

此操作默认通过 CPU 执行，也支持异构加速到 GPU 或 Ascend 上执行。

参数：

data_type (Union[mindspore.dtype, numpy.dtype]) - 指定要转换的数据类型。

异常：

TypeError - 当 data_type 的类型不为 mindspore.dtype 或 numpy.dtype 。

支持平台：

CPU GPU Ascend

样例：

>>> import mindspore.dataset as ds
>>> import mindspore.dataset.transforms as transforms
>>> import numpy as np
>>> from mindspore import dtype as mstype
>>>
>>> # Generate 1d int numpy array from 0 - 63
>>> def generator_1d():
...     for i in range(64):
...         yield (np.array([i]),)
>>>
>>> dataset = ds.GeneratorDataset(generator_1d, column_names='col')
>>> type_cast_op = transforms.TypeCast(mstype.int32)
>>> dataset = dataset.map(operations=type_cast_op)