RandomChoice

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

Apply single transformation randomly picked from a list.

This transform does not support torchscript.

Parameters:

• transforms (sequence or torch.nn.Module) – list of transformations

• p (list of python:floats or None, optional) – probability of each transform being picked. If p doesn’t sum to 1, it is automatically normalized. If None (default), all transforms have the same probability.

Examples using RandomChoice:

How to use CutMix and MixUp

How to use CutMix and MixUp

forward(*inputs: Any) → Any

Do not override this! Use transform() instead.