Pytorch Notes
学习pytorch的一些笔记
Dawn_fire
这个作者很懒,什么都没留下…
展开
-
Pytorch 模型参数管理
Pytorch 模型参数管理1. 参数初始化使用 self.modules ,声明网络时初始化并加载权重pytorch 模型应是 nn.Module 的子类self.modules: nn.Module 类中的一个方法, 返回该网络中的所有 modules可以利用 self.modules 来对网络进行初始化。class Network(nn.Module): def __init__(self): supe().__init__() self.Conv2d = nn.Conv2d原创 2021-12-06 20:41:27 · 1238 阅读 · 0 评论 -
Pytorch 学习笔记 | 01 矩阵的常用操作
文章目录1. pytorch导入2. 矩阵(1)矩阵创建a. 0/1/单位矩阵b. 随机矩阵生成(2)矩阵运算a. 加法b. 访问元素/获取只有一个元素张量的值c. 合并矩阵/拆分矩阵/改变矩阵维度d. 获取size1. pytorch导入from __future__ import print_functionimport torch2. 矩阵(1)矩阵创建torch.tensor...原创 2020-04-02 16:04:51 · 3133 阅读 · 0 评论