LSTM 模型input 参数维度有疑问。

来源:6-6 代码小练(2)

随风漂泊

2019-11-27

老师您好,循环神经网络在pytorch上数据输入维度是(seq_len, batch_size),我之前用过TensorFlow,数据输入维度是(batch_size, seq_len),pytorch这里不理解为什么是相反的。

写回答

1回答

胖虎

2019-11-27

两个框架本身是有些区别的,并不一致

0
4
胖虎
回复
随风漂泊
不客气
2019-11-27
共4条回复

基于Pytorch热门深度学习框架 从零开发NLP聊天机器人

基于中科院所内项目,Seq2Seq+attention主流算法,实战聊天机器人

273 学习 · 107 问题

查看课程