一、 tf.math.round
按元素将张量的值四舍五入到最接近的整数。
import tensorflow as tf
tf.math.round(x, name=None)
四舍五入到偶数。也被称为银行家四舍五入。如果你想根据当前的系统舍入模式进行四舍五入,请使用tf::cint。例如:
x = tf.constant([0.9, 2.5, 2.3, 1.5, -4.5])
tf.round(x) # [ 1.0, 2.0, 2.0, 2.0, -4.0 ]
大白话:小数部分小于或大于0.5,取最近的整数(例如:2.3取2、2.6取3)。小数部分等于0.5时取偶整数(例如:2.5取2、3.5取4)。
二、其他几种取整方式:
1、向上取整
tf.ceil(x, name=None) # 向上取整
2、向下取整
tf.floor(x, name=None) # 向下取整
3、取最接近的整数(实际使用中和tf.round()的取整方式一样)
tf.rint(x, name=None) # 取最接近的整数