Python - tensorflow.math.round()介绍

简介

tensorflow.math.round()是Tensorflow中的一个数学运算函数，用于执行四舍五入运算。

语法

tensorflow.math.round(x, name=None)

参数说明：

• x: 要进行四舍五入运算的张量。
• name: (可选)操作的名称。

返回值：返回一个新的张量，四舍五入后的值与原张量的形状和数据类型相同。

示例

import tensorflow as tf

# 创建一个张量
x = tf.constant([1.4, 2.6, 3.5, 4.9, 5.1], dtype=tf.float32)

# 执行四舍五入运算
result = tf.math.round(x)

# 打印结果
print(result)

输出结果为：

tf.Tensor([1. 3. 4. 5. 5.], shape=(5,), dtype=float32)

总结

tensorflow.math.round()是一个很常用的数学运算函数，可以用于四舍五入运算。在Tensorflow中，它可以使用非常方便，并且返回值与原张量的形状和数据类型相同。