[Numpy-discussion] Consider linking to Accelerate in Apple Silicon ARM64 wheels

2023-03-22 Thread Clemens Brunner
Hello! I recently got a new MacBook Pro with an M2 Pro CPU (ARM64). When I ran some numerical computations (ICA to be precise), I was surprised how slow it was - way slower than e.g. my almost 10 year old Intel Mac. It turns out that the default OpenBLAS, which is what you get when installing t

[Numpy-discussion] Re: broadcasting question

2023-03-22 Thread Robert Kern
On Wed, Mar 22, 2023 at 9:34 AM Neal Becker wrote: > I have a function F > def F(a, b): > c = a * b > > Initially, a is a scalar, b[240,3000]. No problem. > Later I want to use F, where a[240] is a vector. I want to allow both the > scalar and vector cases. So I write: > > def F(a,b): >

[Numpy-discussion] broadcasting question

2023-03-22 Thread Neal Becker
I have a function F def F(a, b): c = a * b Initially, a is a scalar, b[240,3000]. No problem. Later I want to use F, where a[240] is a vector. I want to allow both the scalar and vector cases. So I write: def F(a,b): a = np.atleast_1d(a) c = a[:,None] * b This now works for scalar a o

[Numpy-discussion] Re: Dispatching on `where=` in `__array_ufunc__`?

2023-03-22 Thread Sebastian Berg
On Mon, 2023-03-13 at 11:22 +0100, Sebastian Berg wrote: > Hi all, > > Without concerns voiced, I will probbaly merge the PR: > >     https://github.com/numpy/numpy/pull/23240 I put this PR in, if it turns out there are issues or general dislike, please let us know and we can reconsider. - Seb