I’m sorry for making my problem unclear because of my poor English expression. I appreciate it very much for your reply. But when facing the A = torch.size =(3,3), I couldn’t do change to its parameter linearly unless I do things like this
for i in range(3): for j in range(3): #do change to parameter
And this is nonlinear in my opinion.
But the problem with that is that the code has to change according to torch.size().