pytorch基础变成
文章平均质量分 81
本部分主要讲解pytorch基础编程重要知识点。
奋进的LY
这个作者很懒,什么都没留下…
展开
-
nn.Sequential、nn.ModuleList、nn.ModuleDict区别及使用技巧
nn.ModuleList里面储存了不同 module,并自动将每个 module 的 parameters 添加到网络容器内容(注册),里面的module是按照List的形式顺序存储的,但是在forward中调用的时候可以随意组合。可以任意将 nn.Module 的子类 (比如 nn.Conv2d, nn.Linear 之类的) 加到这个 list 里面,方法和 Python 自带的 list 一样,也就是说它可以使用 extend,append 等操作。原创 2023-06-02 01:08:21 · 810 阅读 · 0 评论 -
torch.matmul()、torch.mul()等函数使用法则
torch.matmul()函数可以用于1维向量×1维向量,1维向量×2维矩阵,2维矩阵×1维向量,矩阵相乘等不同情况。现将相关案例进行介绍方便日后回顾。原创 2023-03-24 20:02:39 · 1496 阅读 · 1 评论