舍入

舍入 - 22

版本

  • 名称: 舍入 (GitHub)

  • 域: main

  • 自版本: 22

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本 22起可用。

摘要

舍入接受一个输入张量并对值进行舍入，逐元素，这意味着它为每个值找到最接近的整数。对于一半的情况，规则是将其舍入到最接近的偶数整数。如果输入 x 是整数、+0、-0、NaN 或无穷大，则返回 x 本身。输出张量与输入具有相同的形状和类型。

示例

round([0.9]) = [1.0]
round([2.5]) = [2.0]
round([2.3]) = [2.0]
round([1.5]) = [2.0]
round([-4.5]) = [-4.0]

输入

  • X (异构) - T

    输入张量

输出

  • Y (异构) - T

    输出张量

类型约束

  • T in ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16) )

    将输入和输出类型约束为浮点张量。

舍入 - 11

版本

  • 名称: 舍入 (GitHub)

  • 域: main

  • 自版本: 11

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本 11起可用。

摘要

舍入接受一个输入张量并对值进行舍入，逐元素，这意味着它为每个值找到最接近的整数。对于一半的情况，规则是将其舍入到最接近的偶数整数。如果输入 x 是整数、+0、-0、NaN 或无穷大，则返回 x 本身。输出张量与输入具有相同的形状和类型。

示例

round([0.9]) = [1.0]
round([2.5]) = [2.0]
round([2.3]) = [2.0]
round([1.5]) = [2.0]
round([-4.5]) = [-4.0]

输入

  • X (异构) - T

    输入张量

输出

  • Y (异构) - T

    输出张量

类型约束

  • T in ( tensor(double), tensor(float), tensor(float16) )

    将输入和输出类型约束为浮点张量。