快捷方式

RandomChoice

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

從列表中隨機選擇並應用單個變換。

此變換不支援 torchscript。

引數:
  • transforms (sequence or torch.nn.Module) – 變換列表

  • p (list of python:floats or None, 可選) – 每個變換被選中的機率。如果 p 的總和不為 1,將自動歸一化。如果為 None(預設值),所有變換具有相同的機率。

使用 RandomChoice 的示例

如何使用 CutMix 和 MixUp

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

不要重寫此方法!請改用 transform()

文件

查閱全面的 PyTorch 開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源