o V h@sPddlmZddlmZmZmZddlZddlZddl m Z Gddde Z dS)) annotations)AnyOptionalUnionN)TVTensorc@s4eZdZdZdddddd dZdddddZdS)Imagea:class:`torch.Tensor` subclass for images with shape ``[..., C, H, W]``. .. note:: In the :ref:`transforms `, ``Image`` instances are largely interchangeable with pure :class:`torch.Tensor`. See :ref:`this note ` for more details. Args: data (tensor-like, PIL.Image.Image): Any data that can be turned into a tensor with :func:`torch.as_tensor` as well as PIL images. dtype (torch.dtype, optional): Desired data type. If omitted, will be inferred from ``data``. device (torch.device, optional): Desired device. If omitted and ``data`` is a :class:`torch.Tensor`, the device is taken from it. Otherwise, the image is constructed on the CPU. requires_grad (bool, optional): Whether autograd should record operations. If omitted and ``data`` is a :class:`torch.Tensor`, the value is taken from it. Otherwise, defaults to ``False``. Ndtypedevice requires_graddatarr Optional[torch.dtype]r 'Optional[Union[torch.device, str, int]]r Optional[bool]returncCsbt|tjjrddlm}||}|j||||d}|jdkr"t|jdkr,| d}| |S)Nr) functionalr ) isinstancePILrtorchvision.transforms.v2r pil_to_tensor _to_tensorndim ValueError unsqueeze as_subclass)clsr r r r Ftensorr Q/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/tv_tensors/_image.py__new__s      z Image.__new__)tensor_contentsr#strcCs|S)N) _make_repr)selfr#r r r!__repr__4szImage.__repr__) r rr rr rr rrr)r#rrr$)__name__ __module__ __qualname____doc__r"r'r r r r!r sr) __future__rtypingrrr PIL.Imagertorch _tv_tensorrrr r r r!s