[Numpy-discussion] new function: broadcast_shapes

[Kevin Sheppard]

There is at least one custom implementation in `Generator` IIRC, so a
formal API addition sounds like a good idea.

Kevin


On Thu, Oct 15, 2020 at 8:52 PM Stephan Hoyer <shoyer at gmail.com> wrote:

> On Thu, Oct 15, 2020 at 11:46 AM Warren Weckesser <
> warren.weckesser at gmail.com> wrote:
>
>> On 10/15/20, Madhulika Jain Chambers <madhulikajain at gmail.com> wrote:
>> > Hello all,
>> >
>> > I opened a PR to add a function which returns the broadcasted shape
>> from a
>> > given set of shapes:
>> > https://github.com/numpy/numpy/pull/17535
>> >
>> > As this is a proposed change to the API, I wanted to see if there was
>> any
>> > feedback from the list.
>>
>>
>> Thanks, this is useful!  I've implemented something similar many times
>> over the years, and could have used it in some SciPy code, where we
>> currently have a private implementation in one of the `stats` modules.
>>
>> Warren
>
>
> +1 for adding this.
>
> There's a version of this helper function -- coincidentally with the
> exactly same API and name -- in JAX, too:
>
> https://github.com/google/jax/blob/22c3684d3bac3ad0f40aca69cdbc76842fa9dfc0/jax/lax/lax.py#L73
>
> _______________________________________________
> NumPy-Discussion mailing list
> NumPy-Discussion at python.org
> https://mail.python.org/mailman/listinfo/numpy-discussion
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mail.python.org/pipermail/numpy-discussion/attachments/20201016/0ebbc50e/attachment.html>