atan2_op

→TensorImplPtr

int atan2_op(const int & y, const int & x)

source edit

Python wrappers

pylucid.ops.atan2function

Quadrant-aware arctangent: $y = \mathrm{atan2}(y_\text{in}, x_\text{in})$ .

Composite over arctan_op plus where_op branches that patch the result with $\pm \pi$ / $\pm \pi/2$ / $0$ corrections based on the signs of the operands. The divide $y / x$ is guarded by substituting $1$ for $x = 0$ so the safe branch is always defined. Gradient flows through the underlying primitives.

Math

\mathrm{atan2}(y, x) = \begin{cases} \arctan(y/x) & x > 0 \\ \arctan(y/x) + \pi & x < 0,\ y \ge 0 \\ \arctan(y/x) - \pi & x < 0,\ y < 0 \\ +\pi/2 & x = 0,\ y > 0 \\ -\pi/2 & x = 0,\ y < 0 \\ 0 & x = 0,\ y = 0 \end{cases}

See Also

arctan_op.

Parameters

yTensorImplPtr

Numerator (sine-like) operand.

xTensorImplPtr

Denominator (cosine-like) operand. Broadcastable with y.

Returns

TensorImplPtr

Angles in $[-\pi, \pi]$ on the broadcast shape of y and x.