tf.keras.ops.square

Return the element-wise square of the input.

Main aliases

tf.keras.ops.numpy.square

x Input tensor.

Output tensor, the square of x.