tf.math.sqrt