human
Here's a good function header given the following:
I wrote the below code

    pad = (logits_size - tf.shape(labels)[1])
    def _PadFn():
        return tf.pad(labels, [[0, 0], [0, pad]])
    def _SliceFn():
        return tf.slice(labels, [0, 0], [(-1), logits_size])
    return tf.cond(tf.greater(pad, 0), _PadFn, _SliceFn)

What's a good function header?
assistant
#Here's a good function header.

def _PadLabels2d(logits_size, labels):