U h@sPddlmZddlmZmZmZddlZddlZddl m Z Gddde Z dS)) annotations)AnyOptionalUnionN)TVTensorc@s0eZdZdZddddddddddd d ZdS) Maska:class:`torch.Tensor` subclass for segmentation and detection masks with shape ``[..., H, W]``. 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 mask 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_gradrzOptional[torch.dtype]z'Optional[Union[torch.device, str, int]]zOptional[bool])datar r r returncCs@t|tjjr$ddlm}||}|j||||d}||S)Nr) functionalr ) isinstancePILZImageZtorchvision.transforms.v2rZ pil_to_tensorZ _to_tensorZ as_subclass)clsr r r r FZtensorrN/var/www/html/venv/lib/python3.8/site-packages/torchvision/tv_tensors/_mask.py__new__s   z Mask.__new__)__name__ __module__ __qualname____doc__rrrrrr s r) __future__rtypingrrrZ PIL.ImagerZtorchZ _tv_tensorrrrrrrs