是否可以整形和np数组并避免不相关的维度1的第一个列表?
array.shape (1,X,X)
to
array2.shape (X,X)
答案 0 :(得分:2)
array = array.reshape(X,X)
-或-
array = array[0]
答案 1 :(得分:0)
只需替换形状:
write
答案 2 :(得分:0)
您可以使用squeeze方法从数组中删除尺寸为1的轴。
array = np.random.randn(1, 10, 10)
array = array.squeeze()
print(array.shape)
# (10, 10)