Функция atan2
Функция atan2 вычисляет арктангенс отношения
y / x, но в отличие от обычного арктангенса
учитывает знаки обоих аргументов, что позволяет
определить правильный квадрант и вернуть угол в
диапазоне от -π до π. Первым параметром
передается значение y (координата по оси Y),
вторым - значение x (координата по оси X).
Оба аргумента могут быть тензорами, числами или
списками.
Синтаксис
tf.math.atan2(y, x, name=None)
Пример
Давайте вычислим арктангенс отношения для простых чисел:
import tensorflow as tf
t = tf.math.atan2(1.0, 1.0)
print(t)
Результат выполнения кода:
tf.Tensor(0.7853982, shape=(), dtype=float32)
Пример
Давайте вычислим арктангенс для тензоров с разными знаками, чтобы увидеть разницу квадрантов:
import tensorflow as tf
y = tf.constant([1.0, -1.0, 1.0, -1.0])
x = tf.constant([1.0, 1.0, -1.0, -1.0])
t = tf.math.atan2(y, x)
print(t)
Результат выполнения кода:
tf.Tensor([ 0.7853982 -0.7853982 2.3561945 -2.3561945], shape=(4,), dtype=float32)
Пример
Давайте вычислим арктангенс для двумерного тензора:
import tensorflow as tf
y = tf.constant([[1.0, 0.0], [0.0, -1.0]])
x = tf.constant([[0.0, 1.0], [-1.0, 0.0]])
t = tf.math.atan2(y, x)
print(t)
Результат выполнения кода:
tf.Tensor(
[[ 1.5707964 0. ]
[ 3.1415927 -1.5707964]], shape=(2, 2), dtype=float32)