torch.select_scatter — PyTorch 2.7 documentation (original) (raw)

torch.select_scatter(input, src, dim, index) → Tensor¶

Embeds the values of the src tensor into input at the given index. This function returns a tensor with fresh storage; it does not create a view.

Parameters

• input (Tensor) – the input tensor.
• src (Tensor) – The tensor to embed into input
• dim (int) – the dimension to insert the slice into.
• index (int) – the index to select with

Note

src must be of the proper size in order to be embedded into input. Specifically, it should have the same shape astorch.select(input, dim, index)

Example:

a = torch.zeros(2, 2) b = torch.ones(2) a.select_scatter(b, 0, 0) tensor([[1., 1.], [0., 0.]])