Indexing

Indexing#

einx.get_at

Retrieves values from a tensor at the coordinates specified by another tensor.

einx.set_at

Sets values in a target tensor from an update tensor at the coordinates specified by an indexing tensor.

einx.add_at

Adds values from an update tensor to a target tensor at the coordinates specified by an indexing tensor.

einx.subtract_at

Subtracts values from a target tensor by an update tensor at the coordinates specified by an indexing tensor.

einx.argmax

Find the coordinates of the maximum values in the given tensor.

einx.argmin

Find the coordinates of the minimum values in the given tensor.

einx.argsort

Returns the indices that would sort values in the given tensor in ascending order.