當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.math.round用法及代碼示例


將張量的值按元素四舍五入到最接近的整數。

用法

tf.math.round(
    x, name=None
)

參數

  • x Tensor 類型為 float16 , float32 , float64 , int32 或 int64 。
  • name 操作的名稱(可選)。

返回

  • 與 x 具有相同形狀和類型的 Tensor 。

四舍五入到一半。也稱為銀行家四舍五入。如果要根據當前係統舍入模式進行舍入，請使用 tf::cint。例如：

x = tf.constant([0.9, 2.5, 2.3, 1.5, -4.5])
tf.round(x)  # [ 1.0, 2.0, 2.0, 2.0, -4.0 ]

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.math.round。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。