plt.imshow（）の画像データの無効な次元
私はkeristバックグラウンドでカプセルネットワークをトレーニングするためにmnistデータセットを使用しています。トレーニング後、mnistデータセットの画像を表示したいと思います。イメージのロードには、mnist.load_data（）が使用されます。データは（x_train、y_train）、（x_test、y_test）として保存されます。今、画像を視覚化するために、私のコードは次のとおりです：
img_path = x_test[1]
print(img_path.shape)
plt.imshow(img_path)
plt.show()
コードは次のように出力します。
(28, 28, 1)
そして次のようにplt.imshow（img_path）のエラー：
TypeError: Invalid dimensions for image data
画像をpng形式で表示する方法。助けて！
テンソルの形状からサイズ1の次元を削除するには、tf.squeezeを使用できます。
plt.imshow( tf.shape( tf.squeeze(x_train) ) )