def METHOD_NAME(self):
    def fn(obj: BaseModelOutput):
        x = obj["last_hidden_state"] * 10
        if "hidden_stats" in obj:
            x += obj["hidden_states"]
        if "attentions" in obj:
            x += obj["attentions"]
        return x
    self._common(fn, 2)