Add a index selected tensor to another tensor with overlapping indices in pytorch

Question

This is a follow up question to this question. I want to do the exactly same thing in pytorch. Is it possible to do this? If yes, how?

import torch
image = to