Skip to content

Commit 6a9da80

Browse files
committed
Document that pts_seconds and duration_seconds are on GPU
1 parent 3e1f4b5 commit 6a9da80

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/torchcodec/_frame.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,10 @@ class FrameBatch(Iterable):
6262
or 5D for sequences of clips, as returned by the :ref:`samplers
6363
<sphx_glr_generated_examples_sampling.py>`. When ``data`` is 4D (resp. 5D)
6464
the ``pts_seconds`` and ``duration_seconds`` tensors are 1D (resp. 2D).
65+
66+
.. note::
67+
The ``pts_seconds`` and ``duration_seconds`` Tensors are always returned
68+
on CPU, even if ``data`` is on GPU.
6569
"""
6670

6771
data: Tensor

0 commit comments

Comments
 (0)