总的来说,除了String类型之外,Python中存在的数据类型,PyTorch中都有对应的数据类型。只不过PyTorch中的数据类型都是Tensor。
PyTorch中的变量是用Variable定义。例如想要定义一个FloatTensor的变量,我们可以这样:先定义一个常量tensor,然后用Variable类包装一下即可:
① w = Variable(torch.tensor([2.0, 3.0]), requires_grad = True)
这个w是一维的,称之为张量。张量可以是一维、二维或多维。如果w需要梯度下降优化,那就要梯度requires_grad = True。如果w不需要梯度下降优化,则设置requires_grad为False。
② b = Variable(torch.tensor(1.0), requires_grad = True)。这个b是0维的,称之为标量。
Variable变量除了有tensor的属性和方法之外,还有一些特殊的属性和方法:grad属性、zero_grad()方法等。
PyTorch的数据的获得
第一行是导入torch包,注意,名字是torch,而不是PyTorch。
第二行代码是引入numpy包,并使用np作为别名。以后代码中不再显示这两行代码。
PyTorch可以从Nu