tanh

nvtripy.tanh(input: Tensor) → Tensor

Computes the elementwise hyperbolic tangent of the elements of the input tensor.

Parameters:

input (Tensor) – [dtype=T1] The input tensor.

Returns:

[dtype=T1] A new tensor.

Return type:

Tensor

DATA TYPE CONSTRAINTS:

• T1: float16, bfloat16, float32, int8

Example

input = tp.arange(3, dtype=tp.float32)
output = tp.tanh(input)

Local Variables

>>> input
tensor([0, 1, 2], dtype=float32, loc=gpu:0, shape=(3,))

>>> output
tensor([0, 0.761594, 0.964028], dtype=float32, loc=gpu:0, shape=(3,))