jax.numpy.cbrt

jax.numpy.cbrt(x)[source]

Return the cube-root of an array, element-wise.

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

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

New in version 1.10.0.

Parameters

x (array_like) – The values whose cube-roots are required.

Returns

y – An array of the same shape as x, containing the cube cube-root of each element in x. If out was provided, y is a reference to it. This is a scalar if x is a scalar.

Return type

ndarray

Examples

>>> np.cbrt([1,8,27])
array([ 1.,  2.,  3.])