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