def METHOD_NAME(x, axis=-1):
    """Perform softmax activation on the data.
    Use approximation to compute exponent for faster speed.
    Parameters
    ----------
    data : tvm.te.Tensor
        can be any dimension
    axis : int
        channel axis
    Returns
    -------
    output : tvm.te.Tensor
        output shape is the same as input
    """
    return softmax_common(x, axis, True)