o €o™iNã@s<ddlmZddlmZddlmZmZGdd„deƒZdS)é)ÚUnion)Ú ImageModule)ÚTensorÚtensorcsDeZdZdZdeeefddf‡fdd„ Zdedefdd „Z‡Z S) ÚRescalezwInitialize the Rescale operator. Args: factor: The scaling factor. Could be a float or a 0-d tensor. ÚfactorÚreturnNcsPtƒ ¡t|tƒrt|ƒ|_dSt|tƒr|jdkr#td|›dƒ‚||_dS)Nrz3Expected factor to be a float or a 0-d tensor, got Ú.) ÚsuperÚ__init__Ú isinstanceÚfloatrrrÚndimÚ TypeError)Úselfr©Ú __class__©úJ/home/ubuntu/.local/lib/python3.10/site-packages/kornia/enhance/rescale.pyr s   zRescale.__init__ÚinputcCs ||jS)N)r)rrrrrÚforward)s zRescale.forward) Ú__name__Ú __module__Ú __qualname__Ú__doc__rr rr rÚ __classcell__rrrrrs rN)ÚtypingrÚ kornia.corerÚModulerrrrrrrÚs