简述

当使用plt时去显示Tensor转出来的np.array时候，遇到这个问题。

这是因为你使用的图是彩图（三维）导致的。

且由于plt显示的时候，对于三维的图片要求的维度是[长，宽，通道数]

解决方案

对于数组处理下对应的维度即：

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

总结

以上是生活随笔为你收集整理的【plt显示Tensor转出来的array时的报错】TypeError: Invalid dimensions for image data的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。