The size of tensor a (3) must match the size of tensor b (51) at non-singleton dimension 3

[kingoftheerath]

此项目我从网上下载的数据和词向量，并构建成了要要求的格式，gen_data.py已经执行成功，但是train过程执行失败，错误内容如下：

请大神帮看看是什么原因，词向量是100维，Python3.7