如何解决输入张量的大小和输入格式不同
我是PyTorch的新手，并与GAN合作开展了一个项目。 我正在尝试使用张量板可视化真实和伪造的图像。
writer_real = SummaryWriter(f"path")
writer_fake = SummaryWriter(f"path")
writer_real.add_image("Temp_real",x_b)
writer_fake.add_image("Temp_fake",fake_b)
在上面的代码中，x_b和fake_b都是张量 但是这段代码给出了一个错误： AssertionError：输入张量的大小和输入格式不同。张量形状：（4，3，128，128），input_format：CHW
我的原始图像是（128x128x3）（HWC）。 请告诉我是否需要其他信息。 请帮忙！
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
