通过atan实现atan2函数

最新推荐文章于 2024-04-29 16:46:08 发布

Yellow Yi

最新推荐文章于 2024-04-29 16:46:08 发布

阅读量1.8k

点赞数 1

分类专栏： pytorch 文章标签： python

本文链接：https://blog.csdn.net/fupotui7870/article/details/116267234

版权

pytorch 专栏收录该内容

5 篇文章

订阅专栏

这篇博客探讨了Python中torch库中atan和atan2函数的使用，特别是当只有atan函数可用时如何模拟atan2的效果。通过示例代码展示了如何通过atan调整象限以得到正确的角度，并对比了atan2的直接输出。内容涵盖了复数的幅角计算和角度在不同象限的校正，对于理解和应用这两个函数具有指导意义。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考MATLAB函数——atan(x)和atan2(x)的区别

在有的情况下只提供了atan的内置函数而没有atan2,先要判断象限再调整。

import torch
import math

pi = math.pi
real = torch.tensor([ 0.12,  -1.1, -0.1, 1])
imag = torch.tensor([ 0.22,  1.2, -1.2, -1.2])
atan = torch.atan(imag/real)
print(atan)
atan[1] = atan[1] + pi
atan[2] = atan[2] - pi
print(atan)
atan2 = torch.atan2(imag,real)
print(atan2)