warp.utils.array_cast#

warp.utils.array_cast(in_array, out_array, count=None)[source]#

Cast elements from one array to another array with a different data type.

This function performs element-wise casting from the input array to the output array. The arrays must have the same number of dimensions and data type shapes. If they don’t match, the arrays will be flattened and casting will be performed at the scalar level.

Parameters:

in_array (array) – Input array to cast from.
out_array (array) – Output array to cast to. Must have the same device as in_array.
count (int | None) – Number of elements to process. If None, processes entire array. For multi-dimensional arrays, partial casting is not supported.

Raises:

RuntimeError – If arrays have different devices, if attempting partial casting on multi-dimensional arrays, if multi-dimensional arrays have different shapes, or if count exceeds the size of either array when the data types differ.

Note

If the input and output arrays have the same data type, this function will simply copy the data without any conversion.