jax.lax.square

内容

jax.lax.square#

jax.lax.square(x)[source]#

元素平方:\(x^2\).

参数::

x (ArrayLike)

返回类型::

Array