tensorflow - tf.round() 到指定的精度

Question

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

有什么办法可以四舍五入到小数点后三位吗？

Answer 1

如果您不冒险达到过高的数字，您可以像这样轻松地做到这一点:

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

注意:x * multiplier 的值不应超过 2^32。所以使用上面的方法，不应该四舍五入太高的数字。