Lambda

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

将用户定义的函数用作转换。

此转换不支持 torchscript。

参数：

lambd (函数) – 用于转换的 Lambda/函数。

extra_repr() → str

设置模块的额外表示。

要打印自定义的额外信息，您应该在自己的模块中重新实现此方法。单行字符串和多行字符串都是可以接受的。