Shortcuts

torch.select_scatter

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 as torch.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.]])

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources