pytorch: 学习笔记2.1:pytorch中的变量 及 tensor(requires_grad=True)转为numpy

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
# 

你可能感兴趣的:(pytorch,python)