pytorch(1) tersor和Variable

版权声明:欢迎大佬指正! https://blog.csdn.net/sinat_36215255/article/details/86310531

莫烦python的教程 terson 与Variable

               

1.Terson 博客

Tensor类似于numpy的ndarrays,另外它还可以在GPU上使用Tensor进行加速计算,其实就是一个多维数组,我们可以对其进行各种数学运算和处理

import torch
x = torch.Tensor(5, 3)
print(x) 
#输出: 
0.0000 0.0000 0.0000 
0.0000 0.4677 0.0000 
0.4692 0.0000 0.4662 
0.0000 0.4693 0.0000 
12.7686 189.2708 0.0000 
[torch.FloatTensor of size 5x3]

terson与numpy是可以相互转化的。

2.Variable博客2

tensor是PyTorch中的完美组件,但是构建神经网络还远远不够,我们需要能够构建计算图的tensor,这就是Variable。Variable是对tensor的封装,操作和tensor是一样的,但是每个Variable都有三个属性,Variable中的tensor本身.data,对应tensor的梯度.grad以及这个Variable是通过说明方式得到的.grad_fn
 

猜你喜欢

转载自blog.csdn.net/sinat_36215255/article/details/86310531
今日推荐