pytorch如何将[0,1]的图像张量转化为[0,255]的张量

可以使用以下代码将[0,1]的图像张量转化为[0,255]的张量：

import torch

img = torch.rand(3, 256, 256)  # 假设图像大小为 3x256x256，像素值在 [0,1] 之间

img = (img * 255).type(torch.uint8)  # 将像素值从 [0,1] 转化为 [0,255]，并将数据类型转化为 uint8

这里使用了 PyTorch 的乘法运算符 `*` 和类型转换函数 `type()`，将像素值从 [0,1] 转化为 [0,255]，并将数据类型从 float 转化为 uint8。