Python – PyTorch atan2() 方法

在 PyTorch 中，atan2() 方法用于返回由给定 Y 和 X 坐标值确定的点的相对于原点的极坐标角度。

语法

torch.atan2(input1, input2, *, out=None) -> Tensor

函数接受以下参数：

• input1 - 输入张量。
• input2 - 输入张量。
• out - 可选的输出张量。

示例

import torch

# 创建两个输入张量
x = torch.tensor([-1.0, 0.0, 1.0])
y = torch.tensor([0.0, 1.0, 1.0])

# 使用atan2()方法计算角度
angle = torch.atan2(y, x)

# 将弧度转换为度
angle = torch.degrees(angle)

print(angle)

输出：

tensor([   90.,    0.,   45.])

在以上示例中，我们首先创建两个输入张量 x 和 y。然后使用 atan2() 计算这些输入张量代表的点的角度。最后，我们将弧度转换为度输出。

结论

在 PyTorch 中，atan2() 方法用于计算由给定 X 和 Y 坐标值确定的点的极坐标角度。它返回以弧度表示的角度值。 如果需要，可以将其转换为度数。