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