JaxBackend.square#
- JaxBackend.square(*args, **kwargs)#
Compute the square of array elements.
- Parameters:
- arrBackendArray
Input data.
- outBackendArray, optional
Output array to write the result to. Returns a new array by default.
- Returns:
- BackendArray
Square of
arr.