[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 保持默认即可。 |