快捷方式

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[source]

不要覆寫這個!請改用 transform()

文件

存取 PyTorch 的完整開發人員文件

檢視文件

教學

取得適合初學者和進階開發人員的深入教學

檢視教學

資源

尋找開發資源並獲得問題解答

檢視資源