Inverse Hyperbolic Tangent (Atanh)#

API#

class warp_nn.modules.operators.Atanh(*, requires_grad: bool = True)[source]#

Bases: Module

Inverse Hyperbolic Tangent (Atanh) operation.

This class computes the element-wise inverse hyperbolic tangent:

\[\text{Atanh}(x) = \frac{1}{2} \ln\left(\frac{1 + x}{1 - x}\right)\]
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.

Raises:

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