jax.numpy.square# jax.numpy.square(x, /)[source]# 返回输入的逐元素平方。 LAX 后端实现的 numpy.square()。 原始文档字符串如下。 参数：: x (array_like) – 输入数据。 返回：: out – 逐元素 x*x，形状和数据类型与 x 相同。如果 x 是标量，则这也是一个标量。 返回类型：: ndarray 或标量
