如何理解未传参的Python包装函数？以PaxML代码为例

先看这段来自PaxML的代码：

def _wrapped_partitioned_step(
        state, prng_key, inputs, unpadded_global_batch_size=None
    ):
    del unpadded_global_batch_size
    return partitioned_step_fn(state, prng_key, inputs)

return _wrapped_partitioned_step, None

疑问解答 #

• 这里返回的是函数对象本身，不是执行函数后的结果。你可以把函数理解成一种特殊的变量，能被返回、传递，后续需要调用这个返回的函数时，才需要传入state、prng_key、inputs这些参数。
• 这个包装函数的核心作用是适配参数签名：调用它的上游代码会传入unpadded_global_batch_size参数，但实际要执行的partitioned_step_fn并不需要这个参数，所以用del unpadded_global_batch_size处理掉这个多余参数，既保证了签名兼容，又不影响原逻辑的执行。

内容的提问来源于stack exchange，提问作者kingwales