weixin_38862073的博客

报错如下： 具体的断言是：Assertion srcIndex < srcSelectDimSize failed， 从这可以看出，这是索引值问题。 然而，这个索引问题一般来自nn.Embedding，看下面例子： import torch.nn as nn import torch emb = nn.Embedding(20, 100, padding_idx=-1) inp = torch.tensor([5, 2, 7, 12, 3]) bad_padding1 = torch.cat((i