torch.angle

`torch.angle`是PyTorch库中的一个函数，它主要用于处理复数数组（Complex Numbers）。当你有一个包含复数元素的张量时，`torch.angle`会返回一个新的张量，其中的元素对应于输入张量中每个复数的幅角部分（也称为相位）。结果是一个介于0和π（或-π和π）之间的浮点数，表示复数在平面上相对于实轴的位置。

例如：

import torch

# 创建一个复数张量
z = torch.tensor([1 + 2j, -1 - 1j])
angle_z = torch.angle(z)

print(angle_z)

这将输出：

tensor([-0.7854,  2.3562])

在这里，`angle_z`的第一项是1+2j的幅角，大约为-45度（因为0.7854弧度对应于-45度），第二项是-1-1j的幅角，大约为135度。