a dH@sddlZddlmZgdZGdddejZGdddejZGdddejZejejd d d Z ejejd d d Z ejejd ddZ dS)N)VflipHflipRot180rot180hflipvflipc@s,eZdZdZejejdddZddZdS)ra1Vertically flip a tensor image or a batch of tensor images. Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The vertically flipped image tensor. Examples: >>> vflip = Vflip() >>> input = torch.tensor([[[ ... [0., 0., 0.], ... [0., 0., 0.], ... [0., 1., 1.] ... ]]]) >>> vflip(input) tensor([[[[0., 1., 1.], [0., 0., 0.], [0., 0., 0.]]]]) inputreturncCst|SN)rselfr rh/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/geometry/transform/flips.pyforwardsz Vflip.forwardcCs|jjSr  __class____name__r rrr__repr__"szVflip.__repr__Nr __module__ __qualname____doc__torchTensorrrrrrrrsrc@s,eZdZdZejejdddZddZdS)ra5Horizontally flip a tensor image or a batch of tensor images. Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The horizontally flipped image tensor. Examples: >>> hflip = Hflip() >>> input = torch.tensor([[[ ... [0., 0., 0.], ... [0., 0., 0.], ... [0., 1., 1.] ... ]]]) >>> hflip(input) tensor([[[[0., 0., 0.], [0., 0., 0.], [1., 1., 0.]]]]) rcCst|Sr )rr rrrr>sz Hflip.forwardcCs|jjSr rrrrrrAszHflip.__repr__Nrrrrrr&src@s,eZdZdZejejdddZddZdS)raRotate a tensor image or a batch of tensor images 180 degrees. Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Examples: >>> rot180 = Rot180() >>> input = torch.tensor([[[ ... [0., 0., 0.], ... [0., 0., 0.], ... [0., 1., 1.] ... ]]]) >>> rot180(input) tensor([[[[1., 1., 0.], [0., 0., 0.], [0., 0., 0.]]]]) rcCst|Sr )rr rrrrZszRot180.forwardcCs|jjSr rrrrrr]szRot180.__repr__NrrrrrrEsrrcCst|ddgS)aRotate a tensor image or a batch of tensor images 180 degrees. .. image:: _static/img/rot180.png Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The rotated image tensor. )rZflip)r rrrrasrcCs*|jd}|dtj|ddd|jdfS)a(Horizontally flip a tensor image or a batch of tensor images. .. image:: _static/img/hflip.png Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The horizontally flipped image tensor. r.deviceshaperZaranger )r wrrrrrs rcCs0|jd}|dtj|ddd|jdddfS)a$Vertically flip a tensor image or a batch of tensor images. .. image:: _static/img/vflip.png Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The vertically flipped image tensor. r.rrrNr!)r hrrrrs r) rZtorch.nnnn__all__Modulerrrrrrrrrrrs