Skip to content

Commit 2ddf90b

Browse files
authored
Document that pts_seconds and duration_seconds are on CPU (#360)
1 parent 3e1f4b5 commit 2ddf90b

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)