model_class.rst

..
  autogenerated from source/_templates/autosummary/model_class.rst

{%- set methods=["forward"] %}
{%- if name in ["Wav2Vec2Model"] %}
  {{ methods.extend(["extract_features"]) }}
{%- elif name in ["Emformer", "RNNTBeamSearch", "WaveRNN", "Tacotron2", ] %}
  {{ methods.extend(["infer"]) }}
{%- elif name == "RNNT" %}
  {{ methods.extend(["transcribe_streaming", "transcribe", "predict", "join"]) }}
{%- endif %}

{{ name | underline }}

.. autoclass:: {{ fullname }}

{% for item in methods %}

{{item | underline("-") }}

.. container:: py attribute

   .. automethod:: {{[fullname, item] | join('.')}}

{%- endfor %}

{%- if name == "RNNTBeamSearch" %}

Support Structures
==================

Hypothesis
----------

.. container:: py attribute

   .. autodata:: torchaudio.models.Hypothesis
      :no-value:

{%- endif %}