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


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


将张量的值按元素四舍五入到最接近的整数。

用法

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

参数

  • x Tensor 类型为 float16 , float32 , float64 , int32int64
  • 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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。