put

→Tensor

put(input: Tensor, index: Tensor, source: Tensor, accumulate: bool = False)

source edit

Write source into input at the flat positions in index.

Mirrors the reference framework's Tensor.put semantics: indices refer to the row-major linearisation of input, regardless of its shape. accumulate=True performs additive scatter (duplicates add), otherwise duplicates resolve to the last write (scatter semantics).

Parameters

inputTensor

Destination — its shape is preserved in the output.

indexTensor

1-D (or flattenable) integer tensor of flat positions in [0, input.numel()).

sourceTensor

Values to scatter; must be flattenable to the same length as index.

accumulatebool= False

If True, add to the existing value at each position; otherwise overwrite.

put