Попрактикуемся с методом reshape. У нас есть трехмерный тензор размерности (6000, 28, 28) . Сопоставьте операцию над этим тензором и её результатат:

К сожалению, у нас пока нет статистики ответов на данный вопрос, но мы работаем над этим.