Identity#

API#

class warp_nn.modules.layers.Identity(*args, **kwargs)[source]#

Identity operation.

This class is placeholder identity operator that is argument-insensitive:

\[\text{Identity}(x) = x\]
Parameters:

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

__call__(
input: array,
) → array[source]#

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.