AssertionError: size of input tensor and input format are different.

AssertionError: size of input tensor and input format are different.         tensor shape: (64, 3, 32, 32), input_format: CHW

原因：tensor的维度是(64, 3, 32, 32)，而要求输入格式是CHW，两者不匹配所以报错。

解决方法：用add_images方法替换add_image方法即可。