diff --git a/src/torchcodec/_frame.py b/src/torchcodec/_frame.py index b9e0fd575..a8fc7f5ba 100644 --- a/src/torchcodec/_frame.py +++ b/src/torchcodec/_frame.py @@ -62,6 +62,10 @@ class FrameBatch(Iterable): or 5D for sequences of clips, as returned by the :ref:`samplers `. When ``data`` is 4D (resp. 5D) the ``pts_seconds`` and ``duration_seconds`` tensors are 1D (resp. 2D). + + .. note:: + The ``pts_seconds`` and ``duration_seconds`` Tensors are always returned + on CPU, even if ``data`` is on GPU. """ data: Tensor