In-place version of torch_index_put. — torch_index_put_ • torch

In-place version of torch_index_put.

Usage

torch_index_put_(self, indices, values, accumulate = FALSE)

Arguments

self: (Tensor) Tensor that will be indexed.
indices: (List[Tensor]) List of indices. Indices are torch tensors with torch_long() dtype.
values: (Tensor) values that will be replaced the indexed location. Used for torch_index_put and torch_index_put_.
accumulate: (bool) Wether instead of replacing the current values with values, you want to add them.