Round

API

class warp_nn.modules.operators.Round(*, requires_grad: bool = True)

Bases: Module

Round operation.

This class computes the element-wise rounding to the nearest integer:

\[\text{Round}(x) = \text{round}(x)\]

Halfway cases are rounded to the nearest even integer (e.g. 0.5 is rounded to 0, and 1.5 to 2).

Parameters:

requires_grad – Whether the cached output arrays of the module require gradients.

__call__(

input: array,

) → array

Forward pass of the operation.

Parameters:

input – The input array, with up to 3 dimensions.

Returns:

The output array, with same shape as the input array.

Raises:

TypeError – If the input array’s data type or number of dimensions is not supported.