Pytorch学习-Variable变量

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