Skip to content

Latest commit

 

History

History
363 lines (287 loc) · 8.88 KB

modules.rst

File metadata and controls

363 lines (287 loc) · 8.88 KB

Modules

ModuleBase

.. autoclass:: texar.torch.ModuleBase
    :members:
    :exclude-members: forward

Embedders

WordEmbedder

.. autoclass:: texar.torch.modules.WordEmbedder
    :members:

PositionEmbedder

.. autoclass:: texar.torch.modules.PositionEmbedder
    :members:

SinusoidsPositionEmbedder

.. autoclass:: texar.torch.modules.SinusoidsPositionEmbedder
    :members:

EmbedderBase

.. autoclass:: texar.torch.modules.EmbedderBase
    :members:
    :exclude-members: forward

Encoders

UnidirectionalRNNEncoder

.. autoclass:: texar.torch.modules.UnidirectionalRNNEncoder
    :members:

BidirectionalRNNEncoder

.. autoclass:: texar.torch.modules.BidirectionalRNNEncoder
    :members:

MultiheadAttentionEncoder

.. autoclass:: texar.torch.modules.MultiheadAttentionEncoder
    :members:

TransformerEncoder

.. autoclass:: texar.torch.modules.TransformerEncoder
    :members:

BERTEncoder

.. autoclass:: texar.torch.modules.BERTEncoder
    :members:

ELMoEncoder

.. autoclass:: texar.torch.modules.ELMoEncoder
    :members:

RoBERTaEncoder

.. autoclass:: texar.torch.modules.RoBERTaEncoder
    :members:

GPT2Encoder

.. autoclass:: texar.torch.modules.GPT2Encoder
    :members:

XLNetEncoder

.. autoclass:: texar.torch.modules.XLNetEncoder
    :members:
    :exclude-members: _forward

Conv1DEncoder

.. autoclass:: texar.torch.modules.Conv1DEncoder
    :members:

EncoderBase

.. autoclass:: texar.torch.modules.EncoderBase
    :members:

RNNEncoderBase

.. autoclass:: texar.torch.modules.RNNEncoderBase
    :members:

default_transformer_poswise_net_hparams

.. autofunction:: texar.torch.modules.default_transformer_poswise_net_hparams

Decoders

.. spelling::
    Luong
    Bahdanau
    Gumbel

DecoderBase

.. autoclass:: texar.torch.modules.DecoderBase
    :members:

RNNDecoderBase

.. autoclass:: texar.torch.modules.RNNDecoderBase
    :members:
    :exclude-members: initialize,step,finalize,output_size

BasicRNNDecoder

.. autoclass:: texar.torch.modules.BasicRNNDecoder
    :members:
    :exclude-members: initialize,step,finalize,output_size

BasicRNNDecoderOutput

.. autoclass:: texar.torch.modules.BasicRNNDecoderOutput
    :members:

AttentionRNNDecoder

.. autoclass:: texar.torch.modules.AttentionRNNDecoder
    :members:
    :exclude-members: initialize,step,finalize,output_size

AttentionRNNDecoderOutput

.. autoclass:: texar.torch.modules.AttentionRNNDecoderOutput
    :members:

GPT2Decoder

.. autoclass:: texar.torch.modules.GPT2Decoder
    :members:

XLNetDecoder

.. autoclass:: texar.torch.modules.XLNetDecoder
    :members:
    :exclude-members: initialize,step,finalize,_create_input

XLNetDecoderOutput

.. autoclass:: texar.torch.modules.XLNetDecoderOutput
    :members:

TransformerDecoder

.. autoclass:: texar.torch.modules.TransformerDecoder
    :members:

TransformerDecoderOutput

.. autoclass:: texar.torch.modules.TransformerDecoderOutput
    :members:

Helper

.. autoclass:: texar.torch.modules.Helper
    :members:

TrainingHelper

.. autoclass:: texar.torch.modules.TrainingHelper
    :members:
    :exclude-members: initialize,sample,next_inputs

EmbeddingHelper

.. autoclass:: texar.torch.modules.EmbeddingHelper
    :members:
    :exclude-members: initialize

GreedyEmbeddingHelper

.. autoclass:: texar.torch.modules.GreedyEmbeddingHelper
    :members:
    :exclude-members: sample

SampleEmbeddingHelper

.. autoclass:: texar.torch.modules.SampleEmbeddingHelper
    :members:
    :exclude-members: sample

TopKSampleEmbeddingHelper

.. autoclass:: texar.torch.modules.TopKSampleEmbeddingHelper
    :members:

TopPSampleEmbeddingHelper

.. autoclass:: texar.torch.modules.TopPSampleEmbeddingHelper
    :members:

SoftmaxEmbeddingHelper

.. autoclass:: texar.torch.modules.SoftmaxEmbeddingHelper
    :members:
    :exclude-members: sample_ids_shape,next_inputs

GumbelSoftmaxEmbeddingHelper

.. autoclass:: texar.torch.modules.GumbelSoftmaxEmbeddingHelper
    :members:

get_helper

.. autofunction:: texar.torch.modules.get_helper

Classifiers

BERTClassifier

.. autoclass:: texar.torch.modules.BERTClassifier
    :members:

RoBERTaClassifier

.. autoclass:: texar.torch.modules.RoBERTaClassifier
    :members:

GPT2Classifier

.. autoclass:: texar.torch.modules.GPT2Classifier
    :members:

Conv1DClassifier

.. autoclass:: texar.torch.modules.Conv1DClassifier
    :members:

XLNetClassifier

.. autoclass:: texar.torch.modules.XLNetClassifier
    :members:

Regressors

XLNetRegressor

.. autoclass:: texar.torch.modules.XLNetRegressor
    :members:

EncoderDecoders

T5EncoderDecoder

.. autoclass:: texar.torch.modules.T5EncoderDecoder
    :members:

Pre-trained

.. spelling::
    pooler

PretrainedMixin

.. autoclass:: texar.torch.modules.PretrainedMixin
    :members:
    :private-members:
    :exclude-members: _name_to_variable

PretrainedBERTMixin

.. autoclass:: texar.torch.modules.PretrainedBERTMixin
    :members:

PretrainedELMoMixin

.. autoclass:: texar.torch.modules.PretrainedELMoMixin
    :members:

PretrainedRoBERTaMixin

.. autoclass:: texar.torch.modules.PretrainedRoBERTaMixin
    :members:

PretrainedGPT2Mixin

.. autoclass:: texar.torch.modules.PretrainedGPT2Mixin
    :members: _init_from_checkpoint

PretrainedXLNetMixin

.. autoclass:: texar.torch.modules.PretrainedXLNetMixin
    :members:

PretrainedT5Mixin

.. autoclass:: texar.torch.modules.PretrainedT5Mixin
    :members:

Connectors

.. spelling::
    reparameterized
    reparameterization
    reparameterizable
    Reparameterization
    Autoencoders
    mlp

ConnectorBase

.. autoclass:: texar.torch.modules.ConnectorBase
    :members:

ConstantConnector

.. autoclass:: texar.torch.modules.ConstantConnector
    :members:

ForwardConnector

.. autoclass:: texar.torch.modules.ForwardConnector
    :members:

MLPTransformConnector

.. autoclass:: texar.torch.modules.MLPTransformConnector
    :members:

Networks

FeedForwardNetworkBase

.. autoclass:: texar.torch.modules.FeedForwardNetworkBase
    :members:

FeedForwardNetwork

.. autoclass:: texar.torch.modules.FeedForwardNetwork
    :members: default_hparams,forward,append_layer,has_layer,layer_by_name,layers_by_name,layers,layer_names

Conv1DNetwork

.. autoclass:: texar.torch.modules.Conv1DNetwork
    :members: default_hparams,forward,append_layer,has_layer,layer_by_name,layers_by_name,layers,layer_names