Как получить правильный массив в многомерном массиве

Question

Python программирование: x_train.shape is (60000,28,28,1) Как я могу преобразовать свои данные в (60000,28,28) Когда я набираю x_train [0], он может показать 28 x 28 массив.

Answer 1

Вы можете использовать numpy.squeeze() для удаления одиночных одномерных записей из формы массива.

np.squeeze(x_train)

Ссылка на документацию