def output_layer(self, features, **kwargs):
        """Project features to the vocabulary size."""
        if self.adaptive_softmax is None:
            # project back to size of vocabulary
            if self.share_input_output_embed:
                return F.linear(features, self.embed_tokens.weight)
            else:
                return F.linear(features, self.embed_out)
        else:
            return features
Project features to the vocabulary size.
understand_icl/fairseq/fairseq/model_parallel/models/pipeline_parallel_transformer/model.py