tf.square()

tf.square(x, name=None)

Computes square of x element-wise.

I.e., (y = x * x = x^2).

Args:

• x: A Tensor or SparseTensor. Must be one of the following types: half, float32, float64, int32, int64, complex64, complex128.
• name: A name for the operation (optional).

Returns:

A Tensor or SparseTensor. Has the same type as x.