import torch
from torch.autograd import Variable
tensor=torch.FloatTensor([[1,2],[3,4]])
Variable=Variable(tensor,requires_grad=True)
#requires_grad能够保存变量的梯度
t_out=torch.mean(tensor*tensor)
V_out=torch.mean(Variable*Variable)
#反向传播变量
V_out.backward()
#打印变量
print(Variable)
#梯度
print(Variable.grad)
#变量转到numpy
print(Variable.data.numpy())
D:\anaconda\envs\pytorch\python.exe D:/pycharm/test/Numpy_torch.py
tensor([[1., 2.],
[3., 4.]], requires_grad=True)
tensor([[0.5000, 1.0000],
[1.5000, 2.0000]])
[[1. 2.]
[3. 4.]]
Process finished with exit code 0