RandomChoice¶
- class torchvision.transforms.v2.RandomChoice(transforms: Sequence[Callable], p: Optional[List[float]] = None)[source]¶
随机应用从列表中选择的单个变换。
此变换不支持 torchscript。
- 参数:
transforms (序列 或 torch.nn.Module) – 变换列表
p (python:floats 列表 或 None, 可选) – 每个变换被选中的概率。如果
p
的总和不为 1,则会自动归一化。如果None
(默认),则所有变换都具有相同的概率。
使用
RandomChoice
的示例