Skip to content

seq2vec_encoder

allennlp.modules.seq2vec_encoders.seq2vec_encoder

[SOURCE]


Seq2VecEncoder#

class Seq2VecEncoder(_EncoderBase,  Registrable)

A Seq2VecEncoder is a Module that takes as input a sequence of vectors and returns a single vector. Input shape : (batch_size, sequence_length, input_dim); output shape: (batch_size, output_dim).

We add two methods to the basic Module API: get_input_dim() and get_output_dim(). You might need this if you want to construct a Linear layer using the output of this encoder, or to raise sensible errors for mis-matching input dimensions.

get_input_dim#

class Seq2VecEncoder(_EncoderBase,  Registrable):
 | ...
 | def get_input_dim(self) -> int

Returns the dimension of the vector input for each element in the sequence input to a Seq2VecEncoder. This is not the shape of the input tensor, but the last element of that shape.

get_output_dim#

class Seq2VecEncoder(_EncoderBase,  Registrable):
 | ...
 | def get_output_dim(self) -> int

Returns the dimension of the final vector output by this Seq2VecEncoder. This is not the shape of the returned tensor, but the last element of that shape.