(x_train, y_train), (x_test, y_test)=datasets.cifar10.load_data()
print((x_train[y_train==5]).shape)
print((x_train[y_train.reshape(-1)==5]).shape)
(x_train, y_train), (x_test, y_test)=datasets.cifar10.load_data()
print((x_train[y_train==5]).shape)
print((x_train[y_train.reshape(-1)==5]).shape)