问题描述
writer.add_image('img/fixed_img', denorm(fixed_img.data), 0)
报如下错误
assert(len(tensor.shape) == len(input_format)), "size of input tensor and input format are different. AssertionError: size of input tensor and input format are different. tensor shape: (128, 3, 64, 64), input_format: CHW
从报错信息来看， input tensor的维度是(128, 3, 64, 64)，而 input_format的格式需要是 CHW。两者不匹配。
分析到这儿就明白了。input tensor虽然格式也是CHW， 但它还有一个batch维度，所以报错。
add_image只接收单一图像，你给它传一个batch数据自然是不行的
解决办法
add_images 替换 add_image
就可以显示batch数据了
