torch (version 0.3.0)

torch_atan2: Atan2

Description

Atan2

Usage

torch_atan2(self, other)

Arguments

self

(Tensor) the first input tensor

other

(Tensor) the second input tensor

atan2(input, other, out=NULL) -> Tensor

Element-wise arctangent of ${\text{input}}_i/{\text{other}}_i$ with consideration of the quadrant. Returns a new tensor with the signed angles in radians between vector $({\text{other}}_i,{\text{input}}_i)$ and vector $(1,0)$. (Note that ${\text{other}}_i$, the second parameter, is the x-coordinate, while ${\text{input}}_i$, the first parameter, is the y-coordinate.)

The shapes of input and other must be broadcastable .

Examples

# NOT RUN {
if (torch_is_installed()) {

a = torch_randn(c(4))
a
torch_atan2(a, torch_randn(c(4)))
}
# }