def fn(*args):
    if ((len(args) == 1) and (not isinstance(args[0], tuple))):
        return batching.padded_batch_window(args[0], padded_shape)
    return tuple([(fn(*arg) if isinstance(arg, tuple) else batching.padded_batch_window(arg, padded_shape)) for arg in args])
def fn(*args):
    if ((len(args) == 1) and (not isinstance(args[0], tuple))):
        return batching.padded_batch_window(args[0], padded_shape)
    return tuple([(fn(*arg) if isinstance(arg, tuple) else batching.padded_batch_window(arg, padded_shape)) for arg in args])