1. 首页
  2. 编程语言
  3. Python
  4. pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换

pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换

上传者: 2022-04-11 18:15:56上传 PDF文件 40.87 KB 热度 14次

1, 创建pytorch 的Tensor张量:2, cpu上的tensor和GPU即pytorch创建的tensor的相互转化3, tensor和numpy的转化4, torch的GPU tensor保存为图片5, 堆叠矩阵,形成彩色图片6, 从numpy数组保存图片7, 读取图片为矩阵:8, 保存矩阵为图片:

用户评论