根据已经给定的维度大小来推断剩余维度的大小
from numpy import random arr = random.randint(30,size=24) #随机的生成24个30以内的整数 print(arr) print("================================") arr = arr.reshape(4,6) print(arr) print("================================") arr = arr.reshape(2,2,6) print(arr) print("================================") arr = arr.reshape(2,2,-1) #自动推断出剩余的一个维度大小为 6 print(arr) print("================================") arr = arr.reshape(2,3,-1) #自动推断出剩余的一个维度大小为 4 print(arr) print("================================") arr = arr.reshape(2,-1,3) #这个 -1 的位置任意 print(arr)结果如下：
[ 7 0 12 12 27 4 17 14 1 20 13 27 15 5 12 12 25 10 0 27 3 15 20 11] ================================ [[ 7 0 12 12 27 4] [17 14 1 20 13 27] [15 5 12 12 25 10] [ 0 27 3 15 20 11]] ================================ [[[ 7 0 12 12 27 4] [17 14 1 20 13 27]] [[15 5 12 12 25 10] [ 0 27 3 15 20 11]]] ================================ [[[ 7 0 12 12 27 4] [17 14 1 20 13 27]] [[15 5 12 12 25 10] [ 0 27 3 15 20 11]]] ================================ [[[ 7 0 12 12] [27 4 17 14] [ 1 20 13 27]] [[15 5 12 12] [25 10 0 27] [ 3 15 20 11]]] ================================ [[[ 7 0 12] [12 27 4] [17 14 1] [20 13 27]] [[15 5 12] [12 25 10] [ 0 27 3] [15 20 11]]]