Me and @pedrohasselmann found a strange issue when using numpy.meshgrid for 3 dimensions and higher. The axis of each grid are shuffled and are not returned in the same order in which they were given.
jax.numpy.meshgrid arg check for "indexing" fails #8495 Closed wztdream opened this issue on Nov 9, 2021 · 1 comment · Fixed by #8498 ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results