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