Функция rsqrt

Функция rsqrt вычисляет обратный квадратный корень поэлементно для переданного тензора. Первым параметром функция принимает тензор или значение, для которого нужно вычислить обратный квадратный корень. Вторым параметром можно передать имя операции.

Математически функция вычисляет значение по формуле 1 / sqrt(x) для каждого элемента тензора. Это часто используется в алгоритмах нормализации и в вычислениях, связанных с расстояниями.

Синтаксис

tf.rsqrt(x, [name])

Пример

Давайте вычислим обратный квадратный корень для тензора чисел 1, 4, 9, 16:

import tensorflow as tf

t = tf.constant([1, 4, 9, 16], dtype=tf.float32)
res = tf.rsqrt(t)
print(res)

Результат выполнения кода:

tf.Tensor([1. 0.5 0.33333334 0.25 ], shape=(4,), dtype=float32)

Пример

Давайте вычислим обратный квадратный корень для двумерного тензора:

import tensorflow as tf

t = tf.constant([[1, 4, 9], [16, 25, 36]], dtype=tf.float32)
res = tf.rsqrt(t)
print(res)

Результат выполнения кода:

tf.Tensor(
[[1.		 0.5		0.33333334]
 [0.25	   0.2		0.16666667]], shape=(2, 3), dtype=float32)

Пример

Давайте получим числовое значение результата с помощью метода numpy:

<+python+> import tensorflow as tf t = tf.constant([1, 4, 9, 16], dtype=tf.float32) res = tf.rsqrt(t) print(res.numpy()) <-python+>

Результат выполнения кода:

[1. 0.5 0.33333334 0.25 ]

Смотрите также

функцию sqrt,
которая вычисляет квадратный корень поэлементно
функцию reciprocal,
которая вычисляет обратное значение поэлементно
функцию pow,
которая возводит тензор в степень
функцию square,
которая вычисляет квадрат поэлементно