pytorch
piupiurui
努力的rui~
展开
-
pytorch学习笔记 —— torch.nn.LSTM
使用 torch.nn.LSTM 可以方便的构建 LSTM,不熟悉 LSTM 的可以先看这两篇文章:RNN:https://blog.csdn.net/yizhishuixiong/article/details/105588233LSTM:https://blog.csdn.net/yizhishuixiong/article/details/105572296下面详细讲述 torch.nn.LSTM 的使用torch.nn.LSTM(input_size, hidden_size, n原创 2020-05-24 23:06:50 · 2598 阅读 · 0 评论 -
pytorch学习笔记 —— torch.nn.Embedding
torch.nn.Embedding 可以实现word embedding,在自然语言处理中比较常用;word embedding 的理解将源数据映射到另外一个空间中,一对一映射。假设将 A 空间中的单词 a1 和 a2 对应 B 空间中的映射是多维变量 b1 和 b2,那么若 a1 <a2 ,则 b1 < b2。通过 word embedding,就可以将自然语言所表示的单词或短语转换为计算机能够理解的由实数构成的向量或矩阵形式(比如,one-hot 就是一种简单的 w..原创 2020-05-24 01:42:37 · 3842 阅读 · 0 评论 -
pytorch学习笔记 —— torch.nn.Module
torch.nn.Module(以下简称 Module)是所有神经网络模块的基类,在 pytorch 中,自定义层、自定义块、自定义模型都可以通过继承 Module 类来实现;Module 中有许多方法,在自定义类时必须重写其中的两个方法:__init__ 和 forward;__init__ 中存放模型的固有属性,如:全连接层、卷积层等等; forward 中写各层之间的连接计算关系,...原创 2020-04-25 13:40:31 · 856 阅读 · 0 评论 -
加载自己的数据集
Dataset类时pytorch图像数据集中最重要的一个类, 是pytorch所有数据集加载应该继承的父类;若要加载自己的数据集,Dataset中的两个私有成员函数必须重新编写: def __getitem__(self, index): def __len__(self):getitem函数:接收的index是一个list的index,这个list的每个元素包含 图片的路...原创 2020-03-31 17:16:04 · 2029 阅读 · 1 评论