input.size(-1) must be equal to input_size. Expected 256, got 131072

img

img

img

img


请问错误原因“input.size(-1) must be equal to input_size. Expected 256, got 131072”该怎么修改

检查输入数据的维度. debug或者print打印一下