UserWarning: nn.init.normal is now deprecated in favor of nn.init.normal_

在保存之前将图像转换为uint8以禁止显示此警告。
解决办法：乘以255，然后加.astype(np.uint8)。