pytorch保存张量为图片

这里用到的是torchvision中的save_image。

废话不多说,直接来代码:

python 复制代码
import torch
from torchvision.utils import save_image
B, C, H, W = 64, 3, 32, 32
input_tensor = torch.randn(B, C, H, W) 

save_image(input_tensor, "hh.png", nrow=8)

首先,这里我们要明确 ( B , C , H , W ) (B, C, H, W) (B,C,H,W)的意思,我们有64张3通道的图片需要可视化。那么,我设定每排放8张。实现效果如图:

参考链接:https://pytorch.org/vision/stable/generated/torchvision.utils.save_image.html

相关推荐
weixin_5142218528 分钟前
FDTD与matlab、python耦合
python·学习·matlab·fdtd
递归不收敛2 小时前
吴恩达机器学习课程(PyTorch 适配)学习笔记大纲
pytorch·学习·机器学习
TGITCIC2 小时前
AI Agent竞争进入下半场:模型只是入场券,系统架构决定胜负
人工智能·ai产品经理·ai产品·ai落地·大模型架构·ai架构·大模型产品
斐夷所非5 小时前
人工智能 AI. 机器学习 ML. 深度学习 DL. 神经网络 NN 的区别与联系
人工智能
递归不收敛5 小时前
吴恩达机器学习课程(PyTorch适配)学习笔记:2.4 激活函数与多类别处理
pytorch·学习·机器学习
F_D_Z5 小时前
数据集相关类代码回顾理解 | StratifiedShuffleSplit\transforms.ToTensor\Counter
python·torchvision·transforms
Funny_AI_LAB6 小时前
OpenAI DevDay 2025:ChatGPT 进化为平台,开启 AI 应用新纪元
人工智能·ai·语言模型·chatgpt
深瞳智检6 小时前
YOLO算法原理详解系列 第002期-YOLOv2 算法原理详解
人工智能·算法·yolo·目标检测·计算机视觉·目标跟踪
tao3556676 小时前
【Python刷力扣hot100】283. Move Zeroes
开发语言·python·leetcode
深眸财经7 小时前
机器人再冲港交所,优艾智合能否破行业困局?
人工智能·机器人