Table Of Contents
Table Of Contents

NDArray.astype

NDArray.astype(dtype, copy=True)[source]

Returns a copy of the array after casting to a specified type.

Parameters:
  • dtype (numpy.dtype or str) – The type of the returned array.
  • copy (bool) – Default True. By default, astype always returns a newly allocated ndarray on the same context. If this is set to False, and the dtype requested is the same as the ndarray’s dtype, the ndarray is returned instead of a copy.
Returns:

The copied array after casting to the specified type, or the same array if copy=False and dtype is the same as the input array.

Return type:

NDArray, CSRNDArray or RowSparseNDArray

Examples

>>> x = mx.nd.zeros((2,3), dtype='float32')
>>> y = x.astype('int32')
>>> y.dtype
<type 'numpy.int32'>