簡述

當使用plt時去顯示Tensor轉出來的np.array時候，遇到這個問題。

這是因為你使用的圖是彩圖（三維）導致的。

且由于plt顯示的時候，對于三維的圖片要求的維度是[長，寬，通道數]

解決方案

對于數組處理下對應的維度即：

array = array.transpose((1, 2, 0)

總結

以上是生活随笔為你收集整理的【plt显示Tensor转出来的array时的报错】TypeError: Invalid dimensions for image data的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。