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

Question

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

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

Answer 1

用numpy去转换，参考下：

Python数组操作将一维数组变成二维数组 - 蔚蓝色の天空 - 博客园 一、问题 我们在进行数组操作的时候会遇到将一个低维的数组变成一个高维的素数组二、解决 第一种方法基本思路就是将低维数组进行等长的循环，在第一次为零的情况下，需要添加一个[]数组，原因是将它的基本框架搭 https://www.cnblogs.com/future-dream/p/10890641.html

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