Переменная
train_images имеет пакет изображений и изображения numpy метрики и срезы работают одинаково для всех метрик в numpy.
Размеры представлены в виде [пакет, строки, столбцы, каналы].
Чтобы получить первое изображение, вы напечатаете: print(train_images[0].shape)
и оно выведет (32, 32, 3).
Чтобы получить первый канал изображения, вы напечатаете: print(train_images[0, :, :, 0])
и оно выведет ( 32, 32) первый канал и т. Д. print(train_images[0, :, :, 1])
для второго канала, print(train_images[0, :, :, 2])
для третьего канала. Где ':' подразумевает все значения.
train_images[0, 0]
будет выводить значения из первой строки первого изображения из пакета (32, 3)
Подробнее: Основы индексации, индексирование массивов