尝试在"weight"中设置形状为 torch.Size([1024, 4096]) 的张量(该张量的形状为 torch.Size([4096, 4096])),这看起来不正确。

—