b3ea6ceefa
* ops to rst * sphinx transition * try html tag * try comment * try code directive * try code directive * try highlight * try concole directive * try line directive * add highlight for code * another directive * introduce consoke directive * add code format
1.8 KiB
Round
@sphinxdirective
Versioned name: Round-5
Category: Arithmetic unary
Short description: Round performs element-wise round operation with given tensor.
Detailed description: Operation takes one input tensor and rounds the values, element-wise, meaning it finds the nearest integer for each value. In case of halves, the rule is to round them to the nearest even integer if mode attribute is half_to_even or rounding in such a way that the result heads away from zero if mode attribute is half_away_from_zero.
.. code-block:: cpp
Input = [-4.5, -1.9, -1.5, 0.5, 0.9, 1.5, 2.3, 2.5]
round(Input, mode = half_to_even) = [-4.0, -2.0, -2.0, 0.0, 1.0, 2.0, 2.0, 2.0]
round(Input, mode = half_away_from_zero) = [-5.0, -2.0, -2.0, 1.0, 1.0, 2.0, 2.0, 3.0]
Attributes:
-
mode
- Description: If set to
half_to_eventhen the rule is to round halves to the nearest even integer, if set tohalf_away_from_zerothen rounding in such a way that the result heads away from zero. - Range of values:
half_to_evenorhalf_away_from_zero - Type: string
- Default value:
half_to_even - Required: no
- Description: If set to
Inputs
- 1: A tensor of type T. Required.
Outputs
- 1: The result of element-wise round operation. A tensor of type T.
Types
- T: any numeric type.
Example
.. code-block:: cpp
<layer ... type="Round">
<data mode="half_to_even"/>
<input>
<port id="0">
<dim>256</dim>
<dim>56</dim>
</port>
</input>
<output>
<port id="1">
<dim>256</dim>
<dim>56</dim>
</port>
</output>
</layer>
@endsphinxdirective