This might be helpful. https://discuss.pytorch.org/t/how-to-assign-a-tensor-with-another-tensor/110644