torch.Tensor.Tensor.index_fill
Fills elements of self at indices along dim with value.
Parameters
dimnumber- Dimension along which to index
indexTensor- 1D tensor with indices
valuenumber- Value to fill
Returns
Tensor<S, D, Dev>– New tensor with filled values