square

rbc.stdlib.elementwise_functions.square(a)

Squares (x_i * x_i) each element x_i of the input array x.

Array-API ‘square’ doc

previous

sqrt

next

subtract