Pytorch model parameter assignment - tensor.copy_ () method

module Intra class state_dict The return object is just a copy, Therefore, modifying the values inside cannot affect the real parameters in the model , We can use tensor.copy_() Method to assign a value .

A simple example ：

a = torch.tensor([[1,2], [3,4]])
b = torch.tensor([[7,8],[9,10]])
a.copy_(b)
print(a)
print(b)