
    3jH                     N    S SK Jr  S SKrS SKJr   " S S\R                  5      rg)    )UnionN)nnc                      ^  \ rS rSrSrS\\\R                  4   SS4U 4S jjr	S\R                  S\R                  4S jr
S	rU =r$ )
Rescale   zqInitialize the Rescale operator.

Args:
    factor: The scaling factor. Could be a float or a 0-d torch.Tensor.

factorreturnNc                   > [         TU ]  5         [        U[        5      (       a  [        R
                  " U5      U l        g [        U[        R                  5      (       a  UR                  S:w  a  [        SU S35      eXl        g )Nr   z9Expected factor to be a float or a 0-d torch.Tensor, got .)
super__init__
isinstancefloattorchtensorr   Tensorndim	TypeError)selfr   	__class__s     P/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/enhance/rescale.pyr   Rescale.__init__    sd    fe$$,,v.DKfell33v{{a7G"[\b[ccd eff K    inputc                     XR                   -  $ )Nr   )r   r   s     r   forwardRescale.forward)   s    {{""r   r   )__name__
__module____qualname____firstlineno____doc__r   r   r   r   r   r   __static_attributes____classcell__)r   s   @r   r   r      sI    !uUELL%89 !d !#U\\ #ell # #r   r   )typingr   r   r   Moduler    r   r   <module>r)      s    $   #bii #r   