def softmax(*, input, dim, dtype=None):
    """
    _stacklevel are ignored here.
    """
    return torch.nn.functional.softmax(input=input, dim=dim, dtype=dtype)
_stacklevel are ignored here.
fx2ait/fx2ait/acc_tracer/acc_ops.py