pytorch错误记录：TypeError: Cannot handle this data type: (1, 1, 64), |u1

如雾如电

已于 2023-03-14 10:01:07 修改

阅读量1.9k

点赞数 2

分类专栏： pytorch 文章标签： pytorch

于 2022-10-21 13:48:20 首次发布

本文链接：https://blog.youkuaiyun.com/qq_20373723/article/details/127444894

版权

pytorch 专栏收录该内容

37 篇文章

订阅专栏

在PyTorch 1.7与1.10/1.11之间的版本变化中，pairwise_distance函数的输出尺寸发生了变化，从(1, 1, 64, 64)变为(1, 64, 64, 1)，这一差异导致了后续图像处理步骤的错误，表现为无法正确转换为PIL图像。解决方案是将PyTorch版本回退到兼容的版本以确保代码正常运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

经过测试：

pairwise_distance函数在pytorch1.7上的表现与pytorch1.10/pytorch1.11上有区别,代码如下

pytorch1.7上

feat_A的size是1，64，64，64，feat_B的size是1，64，64，64

出来应该size是1，1，64，64

而在pytorch1.10/pytorch1.11

出来的size是1，64，64，1

import torch.nn.functional as F

self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True)

直接导致了进一步的计算图像大小存不出来

image_pil = Image.fromarray(image_numpy)
  File "D:\software\Anaconda3\envs\pt3.8\lib\site-packages\PIL\Image.py", line 2815, in fromarray
    raise TypeError("Cannot handle this data type: %s, %s" % typekey) from e
TypeError: Cannot handle this data type: (1, 1, 64), |u1

解决方式只能是版本回退