RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0
参考:12 invalid argument 0: Sizes of tensors must match except in dimension 1. Got 14 and 13 in dimension 0 at /home/prototype/Downloads/pytorch/aten/src/THC/generic/THCTensorMath.cu:83
调用DataSet
的__getitem__
方法返回的image
数据维度应该一致