jax.numpy.ldexp

jax.numpy.ldexp(x1, x2)[source]

Returns x1 * 2**x2, element-wise.

LAX-backend implementation of ldexp(). Original docstring below.

ldexp(x1, x2, /, out=None, *, where=True, casting=’same_kind’, order=’K’, dtype=None, subok=True[, signature, extobj])

The mantissas x1 and twos exponents x2 are used to construct floating point numbers x1 * 2**x2.

Returns

y – The result of x1 * 2**x2. This is a scalar if both x1 and x2 are scalars.

Return type

ndarray or scalar

See also

frexp()

Return (y1, y2) from x = y1 * 2**y2, inverse to ldexp.

Notes

Complex dtypes are not supported, they will raise a TypeError.

ldexp is useful as the inverse of frexp, if used by itself it is more clear to simply use the expression x1 * 2**x2.

Examples

>>> np.ldexp(5, np.arange(4))
array([ 5., 10., 20., 40.], dtype=float16)
>>> x = np.arange(6)
>>> np.ldexp(*np.frexp(x))
array([ 0.,  1.,  2.,  3.,  4.,  5.])