Pytorch入门教程（一）：基本数据类型

1.  python数据类型和pytorch数据类型的对应关系

2.  CPU和GPU下数据类型不同

3  检查数据类型

a = torch.randn(2, 4)  # 创建2行4列随机数组
print(a)
print(a.type())  # 输出数组类型
print(isinstance(a, torch.FloatTensor))  # 输出True/False

输出结果：

tensor([[ 0.0292,  0.9251,  0.5501, -1.1873],
        [-0.9197, -2.6010, -1.5868, -1.3324]])4
torch.FloatTensor
True

4 创建常数 （dimension=0）

a = torch.tensor(2)  # 创建数据
print(a)
print(a.type())  #