src/transformers/models/t5/modeling_tf_t5.py/TFT5ForConditionalGeneration/get_output_embeddings
class TFT5ForConditionalGeneration:    def get_output_embeddings(self):
        if self.config.tie_word_embeddings:
            return self.get_input_embeddings()
        else:
            # in a dense layer the kernel has a shape (last_dim, units), for us (dim, num_tokens)
            # value has a shape (num_tokens, dim) then needs to be transposed
            return tf.transpose(self.lm_head.kernel)