mlx.core.astype

astype(a: array, dtype: Dtype, /, *, stream: None | Stream | Device = None) → array

Cast the array to a specified type.

Parameters:

• a (array) – Input array.

• dtype (Dtype) – Type to which the array is cast.

Returns:

The array with type dtype.

Return type:

array