图像¶
- class torchvision.tv_tensors.Image(data: Any, *, dtype: Optional[dtype] = None, device: Optional[Union[device, str, int]] = None, requires_grad: Optional[bool] = None)[源]¶
torch.Tensor
图像的子类,形状为[..., C, H, W]
。注意
在 转换 中,
Image
实例与纯粹的torch.Tensor
大体上可以互换。有关更多详细信息,请参阅 此注意。- 参数:
data (类张量, PIL.Image.Image) – 任何可以使用
torch.as_tensor()
转换为张量的数据,以及 PIL 图像。dtype (torch.dpython:type, 可选) – 期望的数据类型。如果省略,将从
data
推断。device (torch.device, 可选) – 期望的设备。如果省略且
data
是一个torch.Tensor
,则设备从中获取。否则,图像将在 CPU 上构造。requires_grad (bool, 可选) – Autograd 是否应记录操作。如果省略且
data
是一个torch.Tensor
,则从中取值。否则,默认为False
。
使用
Image
的示例