コード例 #1
0
def hypot(x1, x2):
    _check_arraylike("hypot", x1, x2)
    x1, x2 = _promote_dtypes_inexact(x1, x2)
    x1 = lax.abs(x1)
    x2 = lax.abs(x2)
    x1, x2 = maximum(x1, x2), minimum(x1, x2)
    return lax.select(
        x1 == 0, x1,
        x1 * lax.sqrt(1 + lax.square(
            lax.div(x2, lax.select(x1 == 0, lax_internal._ones(x1), x1)))))
コード例 #2
0
def fmod(x1, x2):
    _check_arraylike("fmod", x1, x2)
    if dtypes.issubdtype(dtypes.result_type(x1, x2), np.integer):
        x2 = _where(x2 == 0, lax_internal._ones(x2), x2)
    return lax.rem(*_promote_args("fmod", x1, x2))