这种情况可能出现在模型的输入与输出的转换过程中。或者模型的上一层输出到下一层输入。
在这些情况下,上一个的输出就是下一个的输入,所以数据的维度是需要对齐(align)的,也就是上一个输出维度需要等于下一个的输入维度。很明显这里的 (24,1) 中的dim1=1维是上一个的输出维度, (3,)中的dim0=3是下一个的输入维度,两者不相等,所以报错。
即:
模型的输出参数维度为3维
但是输入的列维度只有1维,模型的输入参数需要进行reshape,或者填错了
这种情况可能出现在模型的输入与输出的转换过程中。或者模型的上一层输出到下一层输入。
在这些情况下,上一个的输出就是下一个的输入,所以数据的维度是需要对齐(align)的,也就是上一个输出维度需要等于下一个的输入维度。很明显这里的 (24,1) 中的dim1=1维是上一个的输出维度, (3,)中的dim0=3是下一个的输入维度,两者不相等,所以报错。
即:
模型的输出参数维度为3维
但是输入的列维度只有1维,模型的输入参数需要进行reshape,或者填错了