如何在 Pytorch 中将张量转换为复杂类型？

Question

我想用 Pytorch 做一些量子力学计算，其中的数量有时很复杂。我想知道如何将现有的实张量转换为复杂类型。

Answer 1

PyTorch 确实有复数支持。

尝试这个：

import torch

a = torch.tensor([1.0, 2.0], dtype=torch.double)
b = a.type(torch.complex64)