运行代码报错

来源:8-13 建立模型LattentionDecoder

慕尼黑6464164

2020-02-21

LuongAttentionDecoderRNN    forward 方法 
context = torch.bmm(attentionWeight,encoderOutput)
报这个错误:RuntimeError: Expected 3-dimensional tensor, but got 2-dimensional tensor for argument #1 'batch1' (while checking arguments for bmm)

写回答

2回答

胖虎

2020-07-22

torch版本的问题,建议使用1.0或者0.4版本

0
0

胖虎

2020-02-21

这也应该是torch版本的问题,查一下

0
0

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

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

273 学习 · 107 问题

查看课程