tf中小数转化整数的函数

一杯拿铁go

于 2023-02-08 11:55:01 发布

阅读量321

点赞数

分类专栏： TensorFlow&Pytorch 文章标签：深度学习 tensorflow

本文链接：https://blog.youkuaiyun.com/w417950004/article/details/128920650

版权

TensorFlow&Pytorch 专栏收录该内容

4 篇文章

订阅专栏

文章介绍了在TensorFlow中进行数值取整的不同方法，包括tf.round函数的四舍五入规则，tf.ceil用于向上取整，tf.floor用于向下取整，以及tf.rint函数与tf.round类似，都是进行四舍五入但遵循特定的取偶规则。示例代码展示了这些函数在实际操作中的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1，tf.round
四舍五入取偶
小数部分小于0.5，则舍去；
小数部分大于0.5（注：0.5...x 也是大于0.5的），则进位；
小数部分恰好为0.5，取其最靠近的整数。

    with tf.Session() as ss:
        x = tf.constant([0.9, 2.5, 2.3, 1.5, 0.6, 0.51, 0.5, 0.46, -0.5, -4.5])
        RX = tf.round(x)  # [ 1.0, 2.0, 2.0, 2.0, -4.0 ]
        rx2=tf.rint(x, name=None)
        print(RX)
        print(ss.run(RX))
        print(ss.run(rx2))

2，tf.ceil
tf.ceil(x, name=None) # 向上取整
向上取整

3，tf.floor
向下取整
tf.floor(x, name=None) # 向下取整

4，tf.rint
与tf.round一样，四舍五入取偶
tf.rint(x, name=None) # 取最接近的整数