pytorch 基础练习及螺旋数据分类

Posted 2021-10-01 Kerin637

一、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.])

二、螺旋数据分类