tf.keras.ops.square

Return the element-wise square of the input.

x Input tensor.

Output tensor, the square of x.