pytorch的一维数据如(20,)怎么转换为二维的(1,20)

补充:这个一维数据已经是torch里的tensor了,怎么在他是tensor的情况下去变换呢?只使用torch的函数。因为他已经是torch的tensor了,再转去numpy处理不太合适。

nn.Linear需要维度[batch_size,feature],但是我的输入是一维的(20,),会报错

用numpy去转换,参考下:


如果对你有帮助,可以点击我这个回答右上方的【采纳】按钮,给我个采纳吗,谢谢

可用reshape,例如:np.array([1,2,3]).reshape(1,-1)