mlx.core.square

square(a: array, /, *, stream: None | Stream | Device = None) → array

Element-wise square.

參數:

a (array) -- Input array.

回傳:

The square of a.

回傳型別:

array