a d,@s6ddlmZddlZddlmZGdddejZdS))CallableNcs<eZdZdZeddfdd ZejejdddZZ S) LambdaacApplies user-defined lambda as a transform. Args: func: Callable function. Returns: The output of the user-defined lambda. Example: >>> import kornia >>> x = torch.rand(1, 3, 5, 5) >>> f = Lambda(lambda x: kornia.color.rgb_to_grayscale(x)) >>> f(x).shape torch.Size([1, 1, 5, 5]) N)funcreturncs4tt|s*tdtt|j||_dS)Nz'Argument lambd should be callable, got )super__init__callable TypeErrorreprtype__name__r)selfr __class__e/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/contrib/lambda_module.pyrs zLambda.__init__)imgrcOs|j|g|Ri|S)N)r)r rargskwargsrrrforwardszLambda.forward) r __module__ __qualname____doc__rrtorchZTensorr __classcell__rrrrrsr)typingrrZtorch.nnnnModulerrrrrs