Pytorch的多层卷积LSTM(Convolution_LSTM_PyTorch)

Song • 2667 次浏览 • 0 个回复 • 2018年01月30日

Convolution_LSTM_pytorch

多层卷积LSTM模块

卷积式LSTM网络Pytorch实现 :卷积LSTM网络:利用机器学习预测短期降雨

用法

clstm = ConvLSTM(input_channels=512, hidden_channels=[128, 64, 64], kernel_size=5, step=9, effective_step=[2, 4, 8])
lstm_outputs = clstm(cnn_features)
hidden_states = lstm_outputs[0]

项目地址:automan000/Convolution_LSTM_PyTorch


原创文章,转载请注明 :Pytorch的多层卷积LSTM(Convolution_LSTM_PyTorch) - pytorch中文网
原文出处: https://ptorch.com/news/118.html
问题交流群 :168117787
提交评论
要回复文章请先登录注册
用户评论
  • 没有评论
Pytorch是什么?关于Pytorch! PyTorch语义分割(pytorch-semantic-segmentation)