Shortcuts

Lambda

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

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[source]

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.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources