cunumeric.broadcast_shapes

cunumeric.broadcast_shapes(*args: NdShapeLike | Sequence[NdShapeLike]) → NdShape

Broadcast the input shapes into a single shape.

Parameters:

*args (tuples of ints, or ints) – The shapes to be broadcast against each other.

Returns:

tuple

Return type:

Broadcasted shape.

See also

numpy.broadcast_shapes

Availability:

Multiple GPUs, Multiple CPUs