Можно ли изменить форму массива и np и избежать первого списка измерения 1, который не имеет значения?
array.shape (1,X,X) to array2.shape (X,X)
array = array.reshape(X,X)
- или -
array = array[0]
Вы можете удалить ось с размером 1 из массивов, используя метод squeeze .
array = np.random.randn(1, 10, 10) array = array.squeeze() print(array.shape) # (10, 10)
Просто замените форму:
array.shape = array.shape[1:]