Функция sqrt
Функция sqrt вычисляет квадратный корень
из каждого элемента переданного тензора.
Первым параметром функция принимает тензор
или число. Вторым параметром можно передать
имя операции. Функция возвращает тензор
той же формы с поэлементными значениями
квадратного корня.
Синтаксис
tf.sqrt(x, [name])
Пример
Давайте вычислим квадратный корень из тензора
чисел 1, 4, 9, 16, 25:
import tensorflow as tf
t = tf.constant([1, 4, 9, 16, 25])
res = tf.sqrt(t)
print(res)
Результат выполнения кода:
tf.Tensor([1. 2. 3. 4. 5.], shape=(5,), dtype=float32)
Пример
Давайте вычислим квадратный корень из двумерного тензора:
import tensorflow as tf
t = tf.constant([[1, 4, 9], [16, 25, 36]])
res = tf.sqrt(t)
print(res)
Результат выполнения кода:
tf.Tensor(
[[1. 2. 3.]
[4. 5. 6.]], shape=(2, 3), dtype=float32)
Пример
Давайте проверим, что функция sqrt
является обратной к операции возведения
в квадрат:
import tensorflow as tf
t = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0])
res = tf.sqrt(tf.square(t))
print(res)
Результат выполнения кода:
tf.Tensor([1. 2. 3. 4. 5.], shape=(5,), dtype=float32)
Пример
Давайте извлечем значение из тензора
с помощью метода numpy:
import tensorflow as tf
t = tf.constant([4, 9, 16])
res = tf.sqrt(t)
print(res.numpy())
Результат выполнения кода:
[2. 3. 4.]