Python–tensorflow.math.round()
哎哎哎:# t0]https://www.geeksforgeeks.org/python-tensorlow-math-round/
TensorFlow 是谷歌为开发机器学习模型和深度学习神经网络而设计的开源 python 库。 round() 用于寻找元素方向上 x 的最近整数。
语法: tf.math.round(x,name)
参数:
- x: 是输入张量。这个张量允许的数据类型是 bfloat16,half,float32,float64。
- 名称(可选):定义操作的名称。
返回: 返回一个与 x 相同数据类型的张量
例 1:
蟒蛇 3
# 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: 四舍五入到一半。
蟒蛇 3
# 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)