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