pythonmath库round计算_Python tensorflow.math.round()用法及代码示例

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)