RandomChoice — Torchvision 0.22 documentation (original) (raw)

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

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:

forward(*inputs: Any) → Any[source]¶

Do not override this! Use transform() instead.