Функция pow
Функция pow поэлементно возводит первый тензор в степень,
указанную вторым тензором или скаляром.
Первым параметром передаётся основание степени,
вторым - показатель степени.
Оба аргумента могут быть тензорами одинаковой формы
или числами, которые автоматически преобразуются в тензоры.
Синтаксис
tf.pow(x, y, [name])
Пример
Давайте возведём каждый элемент тензора
1, 2, 3, 4, 5 во вторую степень:
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5])
res = tf.pow(t, 2)
print(res)
Результат выполнения кода:
tf.Tensor([ 1 4 9 16 25], shape=(5,), dtype=int32)
Пример
Давайте возведём тензор в степень, заданную другим тензором той же формы:
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5])
exp = tf.constant([1, 2, 3, 2, 1])
res = tf.pow(t, exp)
print(res)
Результат выполнения кода:
tf.Tensor([ 1 4 27 16 5], shape=(5,), dtype=int32)