reshape的时候怎么让行内值不变
时间: 2024-05-06 08:19:58 浏览: 49
TensorFlow的reshape操作 tf.reshape的实现
在使用reshape函数时,你可以使用-1来代表不确定的维度大小。这样,reshape函数会根据其他维度的大小自动计算出该维度的大小,以保证数据总长度不变。例如,如果你有一个(2,3)的数组,想将其reshape为(3,2),可以使用下面的代码:
```
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]])
b = np.reshape(a, (3, -1))
print(b)
```
输出结果为:
```
array([[1, 2],
[3, 4],
[5, 6]])
```
在这个例子中,我们将a reshape为了一个3行,列数不确定的数组。由于a的总长度为6,因此b的列数为6/3=2,从而得到了我们想要的结果。
阅读全文