RandomChoice

class torchvision.transforms.v2.RandomChoice(transforms: Sequence[Callable], p: Optional[List[float]] = None)

从列表中随机选择单个转换并应用。

此转换不支持 torchscript。

参数：

• transforms (序列 或 torch.nn.Module) – 转换列表

• p (python:floats 列表 或 None, 可选) – 选择每个转换的概率。如果 p 的总和不为 1，则会自动进行归一化。如果为 None（默认值），则所有转换的概率相同。

使用 RandomChoice 的示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp

forward(*inputs: Any) → Any

定义每次调用时执行的计算。

所有子类都应覆盖此方法。

注意

虽然前向传递的配方需要在此函数内定义，但应该随后调用 Module 实例，而不是此函数，因为前者负责运行已注册的钩子，而后者则会静默地忽略它们。