一、pytorch 基础练习
主要学习了如何使用torch定义数据以及对数据的一些操作
示例代码中进行点积运算(m @ v)要求数据类型是float,因此前面v定义的时候需要指定dtype,如下所示,
# Create tensor of numbers from 1 to 5
# 注意这里结果是1到4,没有5,即前闭后开
v = torch.arange(1, 5,dtype=torch.float)
print(v)
# tensor([1., 2., 3., 4.])
# Scalar product 点积
# 需要float类型
m @ v
# tensor([30., 60.])