Lambda

class torchvision.transforms.v2.Lambda(lambd: Callable[[Any], Any], *types: Type)

將使用者定義函式用作變換。

此變換不支援 torchscript。

引數：

lambd (function) – 用於變換的 Lambda/函式。

extra_repr() → str

返回模組的額外表示。

要列印定製的額外資訊，您應在自己的模組中重新實現此方法。單行和多行字串均可接受。

transform(inpt: Any, params: Dict[str, Any]) → Any

用於自定義變換的方法。

參閱 如何編寫自己的 v2 變換