Computes square of x element-wise.
tf.math.square(
x, name=None
)
I.e., \(y = x * x = x^2\).
tf.math.square([-2., 0., 3.])
<tf.Tensor: shape=(3,), dtype=float32, numpy=array([4., 0., 9.], dtype=float32)>
Args |
x
|
A Tensor . Must be one of the following types: bfloat16 , half , float32 , float64 , int8 , int16 , int32 , int64 , uint8 , uint16 , uint32 , uint64 , complex64 , complex128 .
|
name
|
A name for the operation (optional).
|
Returns |
A Tensor . Has the same type as x .
If x is a SparseTensor , returns
SparseTensor(x.indices, tf.math.square(x.values, ...), x.dense_shape)
|