JaxBackend.indices

JaxBackend.indices(*args, **kwargs)

Creates an array representing the index grid of an input.

Returns:

BackendArray

The index grid.