o V h@sHddlmZddlmZmZmZddlZddlmZGdddeZ dS)) annotations)AnyOptionalUnionN)TVTensorc@s4eZdZdZdddddd dZdddddZdS)Videoa:class:`torch.Tensor` subclass for videos with shape ``[..., T, C, H, W]``. Args: data (tensor-like): Any data that can be turned into a tensor with :func:`torch.as_tensor`. 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 video 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]returncCs*|j||||d}|jdkrt||S)Nr ) _to_tensorndim ValueError as_subclass)clsr r r r tensorrQ/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/tv_tensors/_video.py__new__s  z Video.__new__)tensor_contentsrstrcCs|S)N) _make_repr)selfrrrr__repr__$szVideo.__repr__) r rr rr rr rrr)rrrr)__name__ __module__ __qualname____doc__rr rrrrr s r) __future__rtypingrrrtorch _tv_tensorrrrrrrs