output = output[:, -1, :] IndexError: too many indices for tensor of dimension 2
时间: 2024-03-04 12:53:38 浏览: 144
PYTHON学习教程:使用list和tuple(详细代码讲解).docx
这个错误是因为你在对一个维度为2的张量进行了过多的索引。通常情况下,输出张量的维度应该是3,第一个维度表示batch size,第二个维度表示序列长度,第三个维度表示每个词的向量表示。在这个错误中,你尝试对一个维度为2的张量进行索引,这是不允许的。你需要检查你的代码,找出哪里出现了这个错误,并修改代码以使维度匹配。
阅读全文