1,pytorch的tensor转为numpy
2,pytorch中的变量Variable
# 1, tensor转为numpy
# requires_grad=True 时,不可以将tensor直接转为numpy; 需要 x(tensor).detach().numpy()
# requires_grad=False 时,可以将tensor直接转为numpy
import torch
from torch.autograd import Variable
x1 = torch.arange(-8.0, 8.0, 1, requires_grad=False) # requires_grad 默认值为False
print(x1.numpy()) # [-8. -7. -6. -5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5. 6. 7.]
x2 = torch.arange(-8.0, 8.0, 1, requires_grad=True)
print(x2.detach().numpy()) # [-8. -7. -6. -5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5. 6. 7.]
# 2, 变量
# pytorch中的变量有 .data & .grad & .grad_fn三个属性
# 定义变量
v1 = Variable(x1, requires_grad=True)
v2 = Variable(x2)
v3 = v1 + v2
# print(variable)
print(v3.data) # 取出tensor数据
print(v3.grad) # 求梯度
print(v3.grad_fn) # 该属性表示此Tensor是不是通过某些运算得到的
# tensor([-16., -14., -12., -10., -8., -6., -4., -2., 0., 2., 4., 6.,
# 8., 10., 12., 14.])
# None
#