mpi_array.globale_ufunc.broadcast_shape

mpi_array.globale_ufunc.broadcast_shape(*shape_args)[source]

Returns the numpy broadcast shape for the give shape arguments.

Parameters:shape2, .. (shape1,) – Array shapes to be broadcast.
Return type:sequence of int
Returns:The broadcast shape.

Examples:

>>> broadcast_shape((4,), (4,))
(4,)
>>> broadcast_shape((4, 1), (1, 5))
(4, 5)
>>> broadcast_shape((4, 1, 3, 7), (1, 8, 1, 7))
(4, 8, 3, 7)
>>> broadcast_shape((3, 7), ())
(3, 7)