mindspore.dataset.transforms.TypeCast
- class mindspore.dataset.transforms.TypeCast(data_type)[源代码]
将输入的Tensor转换为指定的数据类型。
Note
此操作支持通过 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)