Rate this Page
★★★★★

torch.select_scatter#

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

Embeds the values of thesrc tensor intoinput 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 intoinput

  • 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 embeddedintoinput. 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.]])