cr.sparse.arr_largest_index¶
- cr.sparse.arr_largest_index(x)[source]¶
Returns the unraveled index of the largest entry (by magnitude) in an n-d array
- Parameters
x (jax.numpy.ndarray) – An nd-array
- Returns
n-dim index of the largest entry in x
- Return type