JaxBackend.unravel_index

JaxBackend.unravel_index(indices, shape)

Convert flat index to array indices.

Parameters:

indicesBackendArray

Input data.

shapetuple of ints

Shape of the array used for unraveling.

Returns:

BackendArray

Array indices.