torch.Tensor.Tensor.select_scatter
Embeds values from src into self at the given dimension and index.
Parameters
srcTensor- Source tensor to embed
dimnumber- The dimension to select along
indexnumber- The index in the dimension
Returns
Tensor<S, D, Dev>– New tensor with embedded values