问题描述:
File "/data/Code/ceshi.py", line 32, in train_learner
??? now_cls = torch.tensor([[0,1,2,3], [4,5,6]])
ValueError: expected sequence of length 4 at dim 1 (got 3)
问题分析:
????????根据报错信息,问题似乎出在尝试创建一个torch.tensor
对象时,提供的序列的长度不一致。在这里,提供的列表包含两个子列表,而这两个子列表的长度不同。第一个子列表包含4个元素,而第二个子列表只有3个元素。
解决方法:调整使得所有子列表的长度相同,例如:
now_cls = torch.tensor([[0, 1, 2, 3], [4, 5, 6, 7]])