pytorch中的nn.LSTM模块参数详解

直接去官网查看相关信息挺好的，但是为什么有的时候进不去

官网：https://pytorch.org/docs/stable/nn.html#torch.nn.LSTM

使用示例,在使用中解释参数

单向LSTM

import torch
import torch.nn as nnbatch,seq,num_of_feature = (100,10,25)
data = torch.randn(size=(batch,seq,num_of_feature))
lstm = nn.LSTM(input_size=num_of_feature,# ：输入的维度,标量hidden_size=20,#：h的维度,标量num_layers=3,#：堆叠LSTM的层数，默认值为1,标量bias=True,#：偏置 ，默认值：Truebatch_first=True,#： 如果是True，则input为(batch, seq, input_size)。默认值为：False（seq_len, batch, input_size）bidirectional=False,# ：是否双向传播，默认值为False
)output,(h,c) = lstm(data)
print(output.size()) # [batch,seq,hidden_size]
print(h.size()) # [num_layers,batch,hidden_size] 最后一个time_stamp的h状态
print(c.size())# [num_layers,batch,hidden_size] 最后一个time_stamp的c状态

双向LSTM

import torch
import torch.nn as nnbatch,seq,num_of_feature = (100,10,25)
data = torch.randn(size=(batch,seq,num_of_feature))
lstm = nn.LSTM(input_size=num_of_feature,# ：输入的维度,标量hidden_size=20,#：h的维度,标量num_layers=3,#：堆叠LSTM的层数，默认值为1,标量bias=True,#：偏置 ，默认值：Truebatch_first=True,#： 如果是True，则input为(batch, seq, input_size)。默认值为：False（seq_len, batch, input_size）bidirectional=True,# ：是否双向传播，默认值为False
)output,(h,c) = lstm(data)
print(output.size()) # [batch,seq,hidden_size*2]
print(h.size()) # [num_layers*2,batch,hidden_size] 最后一个time_stamp的h状态
print(c.size())# [num_layers*2,batch,hidden_size] 最后一个time_stamp的c状态

参考：https://blog.csdn.net/foneone/article/details/104002372

pytorch中的nn.LSTM模块参数详解相关推荐

Pytorch中dilation（Conv2d）参数详解
目录一.Conv2d 二.Conv2d中的dilation参数一.Conv2d 首先我们看一下Pytorch中的Conv2d的对应函数(Tensor通道排列顺序是:[batch, channel, ...
网络编程中的SO_REUSEADDR和SO_REUSEPORT参数详解
1.SO_REUSEADDR: 在BSD中,SO_REUSEADDR选项有两个用户: 如果有socket绑定了0.0.0.0:port:设置该参数后,其他socket可以绑定本机ip:port.(该功 ...
shell181网格划分_【2017年整理】ANSYS中SHELL181单元理解和参数详解.docx
[2017年整理]ANSYS中SHELL181单元理解和参数详解 ANSYS中SHELL181单元参数详解 SHELL181单元说明: SHELL181单元适合对薄的到具有一定厚度的壳体结构进行分析. ...
matlab中sinks,MATLAB Simulink模块库详解（二）Sinks篇
MATLAB Simulink模块库详解(二)Sinks篇 Simulink模块库概述 1.Sources模块库,为仿真提供各种信号源 2.Sinks模块库,为仿真提供输出设备元件 3.Continu ...
布局中文件中【控件间距参数详解以及单位选择】
注意:例如:android:layout_gravity和android:gravity的区别: android:layout_gravity:此控件相对父控件的相对位置 android:gravit ...
Pytorch LSTM模型参数详解
本文主要依据 Pytorch 中LSTM官方文档,对其中的模型参数.输入.输出进行详细解释. 目录基本原理模型参数 Parameters 输入Inputs: input, (h_0, c_0) 输 ...
python pipline_python中sklearn的pipeline模块实例详解
最近在看<深度学习:基于Keras的Python实践(魏贞原)>这本书,书中8.3创建了一个Scikit-Learn的Pipeline,首先标准化数据集,然后创建和评估基线神经网络模型,代 ...
gini系数决策树_SKlearn中分类决策树的重要参数详解
classsklearn.tree.DecisionTreeClassifier(criterion='gini', splitter='best', max_depth=None, min_samp ...
python requests详解_python的requests模块参数详解
import requests print(dir(requests)) # 1.方法 # ['ConnectTimeout', 'ConnectionError', 'DependencyWarni ...

pytorch中的nn.LSTM模块参数详解

pytorch中的nn.LSTM模块参数详解相关推荐

最新文章

热门文章