tf.round（）到指定的精度

Question

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

有没有办法绕过比方说3位小数？

Answer 1

如果你没有冒太大的风险，你可以轻松地做到这一点：

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

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