Python – tensorflow.math.rint()

TensorFlow是谷歌设计的开源python库，用于开发机器学习模型和深度学习神经网络。rint()用于寻找最接近x的元素整数。

语法:
tf.math.rint(x, name)

参数 :

• x:它是输入的张量。这个张量允许的dtype是bfloat16, half, float32, float64。
• name（可选）:它定义了操作的名称。

返回值:它返回一个与x具有相同 dtype 的张量。

示例 1:

# 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.rint(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：
如果两个值之间存在并列关系，则选择可代表的。

# 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.rint(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)