RuntimeError: beamTokensTable
zchoi opened this issue · 0 comments
zchoi commented
Do we really need transpose？Because I got this error:
beamTokensTable[:, :, 0] = topIdx.transpose(0, 1).data
RuntimeError: The expanded size of the tensor (5) must match the existing size (10) at non-singleton dimension 1. Target sizes: [10, 5]. Tensor sizes: [5, 10]