RFC: zero-dimensional indices array for take #1010

Activity

1. kgryte commented on Aug 6, 2026

   

   kgryte

   on Aug 6, 2026

   Contributor

   More actions

   This was discussed previously. See #855.
2. ev-br commented on Aug 6, 2026

   

   ev-br

   on Aug 6, 2026

   Member

   More actions

   One sticky bit is that the spec requires that

   The output array must have the same number of axes as x

   While major array libraries do not conform for 0D indices:

   In [1]: import jax.numpy as jnp
   
   In [2]: import cupy as cp
   im
   In [3]: import numpy as np
   
   In [4]: for xp in [np, cp, jnp]:
      ...:     x = xp.arange(6).reshape(2, 3)
      ...:     taken = xp.take(x, xp.asarray(0), axis=0)
      ...:     print(xp.__name__, x.ndim, taken.ndim)
      ...: 
   numpy 2 1
   cupy 2 1
   jax.numpy 2 1
   
   
3. mdhaber commented on Aug 7, 2026

   

   mdhaber

   on Aug 7, 2026

   Last edited by mdhaber

   ContributorAuthor

   More actions

   Oops, sorry I missed that issue.
   This is orthogonal: that is only about a 0-d first argument (x), and this is about a 0-d second (indices) argument.
   And @ev-br's point is valid, too.