第27个方法

torch.polar(abs, angle, *, out=None) → Tensor
  • 此方法构造一个复数张量,其元素为与绝对值abs和角angle对应的极坐标所对应的笛卡尔坐标。公式如下:
    在这里插入图片描述
    参数介绍:
  • abs:复数张量的绝对值。必须为float或double。
  • angle:复数张量的角度。数据类型必须与abs相同。
  • out:如果输入为torch.float32,则必须为torch.complex64。如果输入为torch.float64,则必须为torch.complex128。
    使用方法如下:
>>> import numpy as np
>>> abs = torch.tensor([1, 2], dtype=torch.float64)
>>> angle = torch.tensor([np.pi / 2, 5 * np.pi / 4], dtype=torch.float64)
>>> z = torch.polar(abs, angle)
>>> z
tensor([(0.0000+1.0000j), (-1.4142-1.4142j)], dtype=torch.complex128)
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐