Skip to content

output_layer

allennlp.modules.transformer.output_layer

[SOURCE]


OutputLayer

class OutputLayer(TransformerModule,  FromParams):
 | def __init__(
 |     self,
 |     input_size: int,
 |     hidden_size: int,
 |     dropout: float
 | )

get_output_dim

class OutputLayer(TransformerModule,  FromParams):
 | ...
 | def get_output_dim(self) -> int

forward

class OutputLayer(TransformerModule,  FromParams):
 | ...
 | def forward(self, hidden_states, input_tensor)