search

zergtant / pytorch-handbook

pytorch handbook是一本开源的书籍，目标是帮助那些希望和使用PyTorch进行深度学习开发和研究的朋友快速入门，其中包含的Pytorch教程全部通过测试保证可以成功运行
20.1k stars 5.39k forks source link

RNN返回值的另一种写法有问题 #108

Open Shmily1368 opened 3 years ago

Shmily1368 commented 3 years ago

应该更正为: r_out = r_out.view(-1, 64) outs = self.out(r_out) return outs, h_state

zergtant commented 3 years ago

我看下