tv_tensors.rst 701 Bytes
Newer Older
1
.. _tv_tensors:
Nicolas Hug's avatar
Nicolas Hug committed
2

3
TVTensors
Philip Meier's avatar
Philip Meier committed
4
5
==========

6
.. currentmodule:: torchvision.tv_tensors
Nicolas Hug's avatar
Nicolas Hug committed
7

8
9
TVTensors are :class:`torch.Tensor` subclasses which the v2 :ref:`transforms
<transforms>` use under the hood to dispatch their inputs to the appropriate
Nicolas Hug's avatar
Nicolas Hug committed
10
11
12
13
14
15
16
lower-level kernels. Most users do not need to manipulate TVTensors directly.

Refer to
:ref:`sphx_glr_auto_examples_transforms_plot_transforms_getting_started.py` for
an introduction to TVTensors, or
:ref:`sphx_glr_auto_examples_transforms_plot_tv_tensors.py` for more advanced
info.
Nicolas Hug's avatar
Nicolas Hug committed
17

Philip Meier's avatar
Philip Meier committed
18
19
20
21
22
23
24
.. autosummary::
    :toctree: generated/
    :template: class.rst

    Image
    Video
    BoundingBoxFormat
25
    BoundingBoxes
Philip Meier's avatar
Philip Meier committed
26
    Mask
27
    TVTensor
28
    set_return_type
29
    wrap