thtensor专题

RuntimeError: invalid argument 4: out of range at pytorch/torch/lib/TH/generic/THTensor.c:439

使用from torchvision.utils import save_image保存图像时,出现错误: File “/usr/local/lib/python2.7/dist-packages/torchvision/utils.py”, line 51, in save_image grid = make_grid(tensor, nrow=nrow, padding=padding)