dpctl.tensor.broadcast_arrays

dpctl.tensor.broadcast_arrays(*arrays)

Broadcasts one or more dpctl.tensor.usm_ndarrays against one another.

Parameters:

arrays (usm_ndarray) – an arbitrary number of arrays to be broadcasted.

Returns:

A list of broadcasted arrays. Each array must have the same shape. Each array must have the same dtype, device and usm_type attributes as its corresponding input array.

Return type:

List[usm_ndarray]