运行代码报错
来源: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版本
00 -
胖虎
2020-02-21
这也应该是torch版本的问题,查一下
00
相似问题