mindspore.dataset.transforms.TypeCast

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

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

说明

此操作支持通过 Offload 在 Ascend 或 GPU 平台上运行。

参数：

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

异常：

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

支持平台：

CPU Ascend GPU

样例：

>>> 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)