Стать специалистом по машинному обучению

Узнать количество осей тензора train_images можно через атрибут ndim:



print(train_images.ndim)



Форму тензора можно узнать так:



print(train_images.shape)



Тип данных:



print(train_images.dtype)



Вывести 4-ю цифру из этого тензора на экран можно, например, с помощью библиотеки Matplotlib (входит в стандартный набор пакета для научных вычислений)



digit = train_images[4]

import matplotlib.pyplot as plt

plt.imshow(digit, cmap=plt.cm.binary)

plt.show()



У меня код сработал следующим образом: