快捷方式

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 的示例

如何使用 CutMix 和 MixUp

如何使用 CutMix 和 MixUp
forward(*inputs: Any) Any[source]

不要覆盖这个!请改用 transform()

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得您的问题解答

查看资源