src/transformers/models/funnel/modeling_tf_funnel.py/TFFunnelForMaskedLM/serving_output
class TFFunnelForMaskedLM:    def serving_output(self, output: TFMaskedLMOutput) -> TFMaskedLMOutput:
        # hidden_states and attentions not converted to Tensor with tf.convert_to_tensor as they are all of
        # different dimensions
        return TFMaskedLMOutput(logits=output.logits, hidden_states=output.hidden_states, attentions=output.attentions)