jax.numpy.cbrt

jax.numpy.cbrt(x, /)

逐元素返回数组的立方根。

LAX 后端实现 numpy.cbrt()。

原始文档字符串如下。

在版本 1.10.0 中添加。

参数：

x (array_like) – 需要计算立方根的值。

返回值：

y – 与 x 形状相同的数组，包含 x 中每个元素的立方根。如果提供了 out，则 y 是对它的引用。如果 x 是标量，则这是一个标量。

返回类型：

ndarray