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

4
torchaudio.prototype.rnnt_loss
moto's avatar
moto committed
5
==============================
Vincent QB's avatar
Vincent QB committed
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

rnnt_loss
moto's avatar
moto committed
14
~~~~~~~~~
Vincent QB's avatar
Vincent QB committed
15
16
17

.. autofunction:: rnnt_loss

moto's avatar
moto committed
18
19
RNNTLoss
~~~~~~~~
Vincent QB's avatar
Vincent QB committed
20
21
22
23

.. autoclass:: RNNTLoss

  .. automethod:: forward
moto's avatar
moto committed
24
25
26
27
28

References
~~~~~~~~~~

.. footbibliography::