mindspore.dataset.transforms.TypeCast
- class mindspore.dataset.transforms.TypeCast(data_type)[source]
Tensor operation to cast to a given MindSpore data type or NumPy data type.
Note
This operation supports running on Ascend or GPU platforms by Offload.
- Parameters
data_type (Union[mindspore.dtype, numpy.dtype]) – mindspore.dtype or numpy.dtype (e.g.
numpy.float32
) to be cast to.- Raises
TypeError – If data_type is not of MindSpore data type bool, int, float, string or type
numpy.dtype
.
- Supported Platforms:
CPU
Ascend
GPU
Examples
>>> 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)