[AssertionError: nput tensor input format are different]

问题描述

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数据了

[AssertionError: nput tensor input format are different]相关推荐

html input format,html - Input format smpte-timecode - Stack Overflow
Whats the best way to set the input format when using smpte-timecode? placeholder="In" aut ...
UserWarning: This overload of nonzero is deprecated: nonzero(Tensor input, Tensor out) 解决
问题: ../torch/csrc/utils/python_arg_parser.cpp:738: UserWarning: This overload of nonzero is deprecat ...
详解Tensor用法
Tensor的操作如果本文对你有帮助,欢迎点赞.订阅以及star我的项目. 你的支持是我创作的最大动力! 张量的数据属性与 NumPy 数组类似,如下所示: 张量的操作主要包括张量的结构操作和张量的 ...
PyTorch 笔记（06）— Tensor 索引操作（index_select、masked_select、non_zero、gather）
Tensor 支持与 numpy.ndarray 类似的索引操作,如无特殊说明,索引出来的结果与源 tensor 共享内存,即修改一个,另外一个也会跟着改变. In [65]: a = t.arang ...
PyTorch搜索Tensor指定维度的前K大个（K小个）元素--------(torch.topk)命令参数详解及举例
torch.topk 语法 torch.topk(input, k, dim=None, largest=True, sorted=True, *, out = None) 作用返回输入tensor ...
tensor如何实现转置_PyTorch中的傅立叶卷积：通过FFT有效计算大核卷积的数学原理和代码实现...
卷积卷积在数据分析中无处不在.几十年来,它们已用于信号和图像处理.最近,它们已成为现代神经网络的重要组成部分. 在数学上,卷积表示为: 尽管离散卷积在计算应用程序中更为常见,但由于本文使用连续变量证 ...
JavaScript等同于printf / String.Format
我正在寻找一个等效于C / PHP printf()或C#/ Java程序员的String.Format() (适用于.NET的IFormatProvider String.Format() Java ...
Python format（）函数
Python format() function is used to convert given object into string representation based on the pro ...
8.5.4. Interval Input
8.5.4. Interval Input 8.5.4.Interval输入 interval values can be written using the following verbose sy ...

[AssertionError: nput tensor input format are different]

[AssertionError: nput tensor input format are different]相关推荐

最新文章

热门文章