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 is executed on the CPU by default, but it is also supported to be executed on the GPU or Ascend via heterogeneous acceleration.

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 GPU Ascend

Examples

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