ops.rst

torchvision.ops
===============

.. currentmodule:: torchvision.ops

:mod:`torchvision.ops` implements operators that are specific for Computer Vision.

.. note::
  Those operators currently do not support TorchScript.


.. autofunction:: nms
.. autofunction:: roi_align
.. autofunction:: roi_pool

.. autoclass:: RoIAlign
.. autoclass:: RoIPool