其他分享
首页 > 其他分享> > pytorch 基础练习及螺旋数据分类

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

作者:互联网

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

二、螺旋数据分类

标签:练习,torch,tensor,点积,螺旋,dtype,float,pytorch
来源: https://www.cnblogs.com/kerin637/p/15358778.html