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