Используйте torchvision.utils.save_image(), чтобы сохранить тензор как одноканальное изображение в градациях серого.

NoSuchKey

Guess you like

Origin blog.csdn.net/hhb442/article/details/131084052