Source code for torchaug.ta_tensors.nested._video

# ==================================
# Copyright: CEA-LIST/DIASI/SIALV/
# Author : Torchaug Developers
# License: CECILL-C
# ==================================

from __future__ import annotations

from typing import (
    List,
)

from torchaug.ta_tensors import (
    BatchVideos,
    Video,
)

from ._ta_nested_tensors import TANestedTensors



[docs]
class VideoNestedTensors(TANestedTensors[Video, BatchVideos]):
    """Implement Video Nested Tensor for PyTorch."""

    tensors_type = Video
    batch_tensors_type = BatchVideos
    tensors: List[Video]