TensorFlow是Google设计的开源python库,用于开发机器学习模型和深度学习神经网络。 round()用于查找x的逐元素最接近的整数。
用法:tf.math.round(x, name)
参数:
- x:它是输入张量。此张量的允许dtype是bfloat16,half,float32,float64。
- name(optional):它定义了操作的名称。
返回值:
它返回与x相同dtype的张量。
范例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)
相关用法
注:本文由纯净天空筛选整理自aman neekhara大神的英文原创作品 Python – tensorflow.math.round()。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。