Lambda

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

应用用户定义的函数作为转换。

此转换不支持torchscript。

Parameters:

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

extra_repr() → str

设置模块的额外表示。

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