tf.math.sqrt

Computes element-wise square root of the input tensor.

View aliases

Main aliases

tf.sqrt

Compat aliases for migration

tf.compat.v1.sqrt

tf.math.sqrt(
    x, name=None
)

Used in the notebooks

Used in the guide	Used in the tutorials
Distributed training with Core APIs and DTensor Multilayer perceptrons for digit recognition with Core APIs Optimizers with Core APIs	Uncertainty-aware Deep Learning with SNGP TFP Release Notes notebook (0.12.1) Copulas Primer Substantial Undocumented Infection Facilitates the Rapid Dissemination of Novel Coronavirus (SARS-CoV2) Neural machine translation with a Transformer and Keras

Note: This operation does not support integer types.

x = tf.constant([[4.0], [16.0]])
tf.sqrt(x)
<tf.Tensor: shape=(2, 1), dtype=float32, numpy=
  array([[2.],
         [4.]], dtype=float32)>
y = tf.constant([[-4.0], [16.0]])
tf.sqrt(y)
<tf.Tensor: shape=(2, 1), dtype=float32, numpy=
  array([[nan],
         [ 4.]], dtype=float32)>
z = tf.constant([[-1.0], [16.0]], dtype=tf.complex128)
tf.sqrt(z)
<tf.Tensor: shape=(2, 1), dtype=complex128, numpy=
  array([[0.0+1.j],
         [4.0+0.j]])>

Note: In order to support complex type, please provide an input tensor of complex64 or complex128.

Args
`x`	A `tf.Tensor` of type `bfloat16`, `half`, `float32`, `float64`, `complex64`, `complex128`
`name`	A name for the operation (optional).

Returns
A `tf.Tensor` of same size, type and sparsity as `x`. If `x` is a `SparseTensor`, returns `SparseTensor(x.indices, tf.math.sqrt(x.values, ...), x.dense_shape)`

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/math/sqrt