jax.numpy.cbrt

jax.numpy.cbrt(x, /)

计算输入数组的逐元素立方根。

numpy.cbrt 的 JAX 实现。

参数:

x (ArrayLike) – 输入数组或标量。complex 数据类型不支持。

返回:

一个包含 x 元素立方根的数组。

返回类型:

数组

另请参阅

• jax.numpy.sqrt(): 计算输入的逐元素非负平方根。

• jax.numpy.square(): 计算输入的逐元素平方。

示例

>>> x = jnp.array([[216, 125, 64],
...                [-27, -8, -1]])
>>> with jnp.printoptions(precision=3, suppress=True):
...   jnp.cbrt(x)
Array([[ 6.,  5.,  4.],
       [-3., -2., -1.]], dtype=float32)