jax.lax.cbrt# jax.lax.cbrt(x)[source]# Elementwise cube root: \(\sqrt[3]{x}\). Parameters: x (ArrayLike) Return type: Array
