Aggiungi un tensore selezionato all'indice a un altro tensore con indici sovrapposti in una torcia
Questa è una domanda successiva a questa domanda . Voglio fare esattamente la stessa cosa con Pytorch. È possibile farlo? Se sì, come?
import torch
image = torch.tensor([[246, 50, 101], [116, 1, 113], [187, 110, 64]])
iy = torch.tensor([[1, 0, 2], [1, 0, 2], [2, 2, 2]])
ix = torch.tensor([[0, 2, 1], [1, 2, 0], [0, 1, 2]])
warped_image = torch.zeros(size=image.shape)
Ho bisogno di qualcosa del genere torch.add.at(warped_image, (iy, ix), image)
che dia l'output come
[[ 0. 0. 51.]
[246. 116. 0.]
[300. 211. 64.]]
Notare che gli indici in (0,1)
e (1,1)
puntano alla stessa posizione (0,2)
. Quindi, io voglio warped_image[0,2] = image[0,1] + image[1,1] = 51
.
Risposte
3 Ivan
Quello che stai cercando è torch.Tensor.index_put_con l' accumulate
argomento impostato su True
:
>>> warped_image = torch.zeros_like(image)
>>> warped_image.index_put_((iy, ix), image, accumulate=True)
tensor([[ 0, 0, 51],
[246, 116, 0],
[300, 211, 64]])
Oppure, utilizzando la versione out-place torch.index_put
:
>>> torch.index_put(torch.zeros_like(image), (iy, ix), image, accumulate=True)
tensor([[ 0, 0, 51],
[246, 116, 0],
[300, 211, 64]])