Lambda

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

Apply a user-defined function as a transform.

This transform does not support torchscript.

Parameters:

lambd (function) – Lambda/function to be used for transform.

extra_repr() → str

Set the extra representation of the module.

To print customized extra information, you should re-implement this method in your own modules. Both single-line and multi-line strings are acceptable.