我在執(zhí)行 tf.round(x) x=[0.4, 1.5, 2.5, -1.5, -2.5, -0.4] 時(shí)遇到一些問題如果我想讓 ans=[0, 2, 3, -2, -3, 0] 從零舍入一半我應(yīng)該怎么做？我嘗試過 tf.keras.backend.round()、tf.math.round、tf.math.rint()我在 python 中得到了類似的答案，但在 TF 中沒有得到類似的答案>>>decimal.Decimal(101.5).quantize(decimal.Decimal('0'), rounding=decimal.ROUND_HALF_UP)Decimal('102')>>>decimal.Decimal(102.5).quantize(decimal.Decimal('0'), rounding=decimal.ROUND_HALF_UP)Decimal('103')>>>decimal.Decimal(-101.5).quantize(decimal.Decimal('0'), rounding=decimal.ROUND_HALF_UP)Decimal('-102')>>>decimal.Decimal(-102.5).quantize(decimal.Decimal('0'), rounding=decimal.ROUND_HALF_UP)Decimal('-103')謝謝
2 回答
手掌心
TA貢獻(xiàn)1942條經(jīng)驗(yàn) 獲得超3個(gè)贊
嘗試
x=tf.constant([0.4, 1.5, 2.5, -1.5, -2.5, -0.4]) x=tf.where(x>0, tf.math.nextafter(x, np.inf), tf.math.nextafter(x, -np.inf)) x=tf.round(x)
這將從 0 開始舍入。
炎炎設(shè)計(jì)
TA貢獻(xiàn)1808條經(jīng)驗(yàn) 獲得超4個(gè)贊
這個(gè)怎么樣 ？
x = np.array([0.4, 1.5, 2.5, -1.5, -2.5, -0.4])
for i, val in enumerate(x):
if val % 1 == 0.5
x[i] = tf.math.floor(x[i]) if val < 0 else tf.math.floor(x[i]+0.5)
else
x[i] = tf.math.round(x[i])
print(x)
[ 0. 2. 3. -2. -3. 0.]
添加回答
舉報(bào)
0/150
提交
取消