PytorchBackend.indices#

PytorchBackend.indices(shape)[source]#

Creates an array representing the index grid of an input.

Returns:
BackendArray

The index grid.

On this page
Show Source