当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tensorflow.math.rint()用法及代码示例


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

用法:tf.math.rint(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.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:如果两个值之间存在平局,甚至可以选择代表。

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.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)



相关用法


注:本文由纯净天空筛选整理自aman neekhara大神的英文原创作品 Python – tensorflow.math.rint()。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。