Pytorch LSTM函数参数解释 图解
LSTM函数解释 pytorch版本torch.nn.LSTM函数图解LSTM函数引用图片笔者最近在写有关LSTM的代码,但是对于nn.LSTM函数中的有些参数还是不明白其具体含义,学习过后在此记录。为了方便说明,我们先解释函数参数的作用,接着对应图片来说明每个参数的具体含义。torch.nn.LSTM函数LSTM的函数class torch.nn.LSTM(args, *kwargs) # 主要参数 # input_size – 输入的特征维度 # hidden_size – 隐状态的特征
原创
2021-03-10 17:28:54 ·
3213 阅读 ·
2 评论