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()。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。