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 变换