jax.numpy.exp

Contents

jax.numpy.exp#

jax.numpy.exp(x, /)#

Calculate the exponential of all elements in the input array.

LAX-backend implementation of numpy.exp().

Original docstring below.

Parameters:

x (array_like) – Input values.

Returns:

out – Output array, element-wise exponential of x. This is a scalar if x is a scalar.

Return type:

ndarray or scalar

References