Python – tensorflow.math.round()

TensorFlow 是由 Google 设计的开源Python库，用于开发机器学习模型和深度学习神经网络。 round()用于查找 x 的按元素最近的整数。

Syntax: tf.math.round(x, name)

Parameter:

• x: It’s the input tensor. Allowed dtype for this tensor are bfloat16, half, float32, float64.
• name(optional): It’s defines the name for the operation.

Returns:
It returns a tensor of same dtype as x.

示例 1：

Python3

# Importing the library
import tensorflow as tf
  
# Initializing the input tensor
a = tf.constant([.2, 1.6, 3.7, 1], dtype = tf.float64)
  
# Printing the input tensor
print('a: ', a)
  
# Calculating result
res = tf.math.round(a)
  
# Printing the result
print('Result: ', res)

输出：

a:  tf.Tensor([0.2 1.6 3.7 1. ], shape=(4, ), dtype=float64)
Result:  tf.Tensor([0. 2. 4. 1.], shape=(4, ), dtype=float64)

示例 2：四舍五入到偶数。

Python3

# importing the library
import tensorflow as tf
  
# Initializing the input tensor
a = tf.constant([2.5, 5.5, 1.5, 6.5], dtype = tf.float64)
  
# Printing the input tensor
print('a: ', a)
  
# Calculating result
res = tf.math.round(a)
  
# Printing the result
print('Result: ', res)

输出：

a:  tf.Tensor([2.5 5.5 1.5 6.5], shape=(4, ), dtype=float64)
Result:  tf.Tensor([2. 6. 2. 6.], shape=(4, ), dtype=float64)