MLXBackend.indices#

MLXBackend.indices(arr)[source]#

Creates an array representing the index grid of an input.

Returns:
BackendArray

The index grid.

On this page
Show Source