快捷方式

Lambda

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

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

此变换不支持 torchscript。

参数:

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

extra_repr() str[source]

返回模块的额外表示。

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

transform(inpt: Any, params: Dict[str, Any]) Any[source]

用于为自定义变换重写的方法。

请参阅 如何编写您自己的 v2 变换

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源