jax.numpy.sqrt# jax.numpy.sqrt(x, /)[source]# 逐元素返回数组的非负平方根。 LAX 后端实现 numpy.sqrt()。 原始文档字符串如下。 参数： x (array_like) – 需要计算平方根的值。 返回值： y – 与 x 形状相同的数组，包含 x 中每个元素的正平方根。如果 x 中的任何元素是复数，则返回一个复数数组（并计算负实数的平方根）。如果 x 中的所有元素都是实数，则 y 也是实数，负元素返回 nan。如果提供了 out，则 y 是对它的引用。如果 x 是标量，则这是一个标量。 返回类型： ndarray
