tf.round（）到指定的精度

Posted 2021-03-31

tf.round(x)将x的值舍入为整数值。

是否有任何方法可以舍入到3位小数位？

答案

如果你不冒险达到太高的数字，你可以轻松地做到这一点：

def my_tf_round(x, decimals = 0):
    multiplier = tf.constant(10**decimals, dtype=x.dtype)
    return tf.round(x * multiplier) / multiplier

提及：x *乘数的值不应超过2 ^ 32。所以使用上面的方法，不应该舍得太高的数字。