pytorch - 如何将张量转换为 Pytorch 中的复杂类型？

Question

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

标签： pytorch

Answer 1

PyTorch 确实支持复数。

尝试这个：

import torch

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