[paddle 参数更多]torchvision.transforms.ToTensor

torchvision.transforms.ToTensor

torchvision.transforms.ToTensor()

paddle.vision.transforms.ToTensor

paddle.vision.transforms.ToTensor(data_format: str = 'CHW', keys: List[str] | Tuple[str] = None)

两者功能基本一致,但 Paddle 相比 torchvision 支持更多参数,具体如下:

参数映射

torchvision PaddlePaddle 备注
- data_format 返回 Tensor 的格式,PyTorch 无此参数,Paddle 保持默认即可。
- keys 输入的类型,PyTorch 无此参数,Paddle 保持默认即可。