TerryChan' Blog

文章

标签

喜欢

标签: lstm

共找到 7 篇相关文章

第 1 页,共 1 页

文章预览

nn Sequential 中使用LSTM/GRU
2023年08月22日 421 字

由于LSTM/GRU这种是多个输出,所以没法直接使用,可以采用自定义模块的方式来转化成一个。 还有一个方案: ``` # I made a module called SelectItem to pick out an element from a tuple or list class SelectItem(nn.Module): d…

阅读全文
浏览 评论

文章预览

LSTM+Attention demo
2023年08月22日 431 字

``` #!/usr/bin/python # -*- coding: utf-8 -*- import torch from torch import nn import torch.nn.functional as F # 定义 单双向LSTM 模型 class Rnn(nn.Module): ”“” 双向lstm “”“ …

阅读全文
浏览 评论