torch_atan2 function

Atan2

Atan2

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 \mboxinputi/\mboxotheri\mbox{input}_{i} / \mbox{other}_{i}

with consideration of the quadrant. Returns a new tensor with the signed angles in radians between vector (\mboxotheri,\mboxinputi)(\mbox{other}_{i}, \mbox{input}_{i})

and vector (1,0)(1, 0). (Note that \mboxotheri\mbox{other}_{i}, the second parameter, is the x-coordinate, while \mboxinputi\mbox{input}_{i}, the first parameter, is the y-coordinate.)

The shapes of input and other must be broadcastable .

Examples

if (torch_is_installed()) { a = torch_randn(c(4)) a torch_atan2(a, torch_randn(c(4))) }
  • Maintainer: Daniel Falbel
  • License: MIT + file LICENSE
  • Last published: 2025-02-14