神经网络里面的参数都是变量(variable)的形式。
Tensor和Variable在反向传播的时候会出现差别,Tensor不能反向传播,Variable可以反向传播。
Variable containing是指该variable属于搭建图的一部分。
示例代码1:
import torch
from torch.autograd import Variable
tensor = torch.FloatTensor([[1, 2], [3, 4]])
# requires_grad表示需不需要将定义的variable放到反向传播中
variable = Variable(tensor, requires_grad=True)
print(tensor)
print(variable)
运行结果:
1 2
3 4
[torch.FloatTensor of size 2x2]Variable containing:
1 2
3 4
[torch.FloatTensor of size 2x2]