Function round

imperative/python/megengine/functional/elemwise.py:852–887

r"""Element-wise :math:`\operatorname{round}(x)` function. Rounds each element :math:`x_i` of the input tensor :math:`x` to the nearest integer-valued number. Args: x: input tensor. Should have a numeric data type. Returns: a tensor containing the rounded resul

(x)

def round(x):
    r"""Element-wise :math:`\operatorname{round}(x)` function.
    
    Rounds each element :math:`x_i` of the input tensor :math:`x` to the nearest integer-valued number.
    
    Args:
        x: input tensor. Should have a numeric data type.

    Returns:
        a tensor containing the rounded result for each element in :math:`x`.
        The returned tensor must have the same data type as :math:`x`.

    .. admonition:: Special cases

       If :math:`x_i` is already integer-valued, the result is :math:`x_i`.
    
       For floating-point operands,

       * If :math:`x_i` is ``+infinity``, the result is ``+infinity``.
       * If :math:`x_i` is ``-infinity``, the result is ``-infinity``.
       * If :math:`x_i` is ``+0``, the result is ``+0``.
       * If :math:`x_i` is ``-0``, the result is ``-0``.
       * If :math:`x_i` is NaN, the result is NaN.
       * If two integers are equally close to :math:`x_i`, the result is the even integer closest to :math:`x_i`.

    Examples:
        >>> F.round(1.5)
        Tensor(2.0, device=xpux:0)

        Element-wise rounding:

        >>> x = Tensor([1.5, 2.5, 3.5, 4.5])
        >>> F.round(x)
        Tensor([2. 3. 4. 5.], device=xpux:0)
    """
    return _elwise(x, mode=Elemwise.Mode.ROUND)


def ceil(x):