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

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)