rnnt_loss.rst 605 Bytes
Newer Older
Vincent QB's avatar
Vincent QB committed
1
2
3
.. role:: hidden
    :class: hidden-section

4
torchaudio.prototype.rnnt_loss
Vincent QB's avatar
Vincent QB committed
5
6
===============================

7
.. currentmodule:: torchaudio.prototype.rnnt_loss
Vincent QB's avatar
Vincent QB committed
8
9
10

.. note::

11
    The RNN transducer loss is a prototype feature, see `here <https://pytorch.org/audio>`_ to learn more about the nomenclature. It is only available within the nightlies, and also needs to be imported explicitly using: :code:`from torchaudio.prototype.rnnt_loss import rnnt_loss, RNNTLoss`.
Vincent QB's avatar
Vincent QB committed
12
13
14
15
16
17
18
19
20
21
22
23

rnnt_loss
---------

.. autofunction:: rnnt_loss

:hidden:`RNNTLoss`
~~~~~~~~~~~~~~~~~~

.. autoclass:: RNNTLoss

  .. automethod:: forward