补充:这个一维数据已经是torch里的tensor了,怎么在他是tensor的情况下去变换呢?只使用torch的函数。因为他已经是torch的tensor了,再转去numpy处理不太合适。
nn.Linear需要维度[batch_size,feature],但是我的输入是一维的(20,),会报错
用numpy去转换,参考下:
可用reshape,例如:np.array([1,2,3]).reshape(1,-1)