tf.math.atan2

Поэлементно вычисляет арктангенс y/x с учетом знаков аргументов.

tf.math.atan2(
    y: Annotated[Any, tf.raw_ops.Any],
    x: Annotated[Any, tf.raw_ops.Any],
    name=None
) -> Annotated[Any, tf.raw_ops.Any]

Используется в блокнотах

Это угол \( \theta \in [-\pi, \pi] \), для которого выполняются равенства \[ x = r \cos(\theta) \] и \[ y = r \sin(\theta) \], где \(r = \sqrt{x^2 + y^2} \).

Например:

x = [1., 1.]
y = [1., -1.]
print((tf.math.atan2(y,x) * (180 / np.pi)).numpy())
[ 45. -45.]