tf.round() à une précision spécifiée

Question

tf.round(x) arrondit les valeurs de x à des valeurs entières.

Existe-t-il un moyen d'arrondir à, disons, 3 décimales à la place?

Answer 1

Vous pouvez le faire facilement comme ça, si vous ne risquez pas d'atteindre un nombre trop élevé:

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