torch_index_put_ function

In-place version of torch_index_put.

In-place version of torch_index_put.

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.
  • Maintainer: Daniel Falbel
  • License: MIT + file LICENSE
  • Last published: 2025-02-14