Функция sign

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

Синтаксис

tf.sign(x, [name])

Пример

Давайте применим функцию sign к тензору из чисел -3, -1, 0, 2, 5:

import tensorflow as tf

t = tf.constant([-3, -1, 0, 2, 5])
res = tf.sign(t)
print(res)

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

tf.Tensor([-1 -1 0 1 1], shape=(5,), dtype=int32)

Пример

Давайте применим функцию sign к тензору с типом float32, содержащему отрицательные и положительные значения:

import tensorflow as tf

t = tf.constant([-2.5, 0.0, 3.7], dtype=tf.float32)
res = tf.sign(t)
print(res)

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

tf.Tensor([-1. 0. 1.], shape=(3,), dtype=float32)

Пример

Давайте применим функцию sign к двумерному тензору:

import tensorflow as tf

t = tf.constant([[-1, 2, -3], [4, 0, -6]])
res = tf.sign(t)
print(res)

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

tf.Tensor(
[[-1  1 -1]
 [ 1  0 -1]], shape=(2, 3), dtype=int32)

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

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