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

Question

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

Answer 1

PyTorch确实有complex number support。

尝试一下：

import torch

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

Answer 2

AFAIK pytorch不支持复杂的张量。
您可以使用两个实数张量表示一个复杂的张量。但是，您将必须实现所有相关的基本算术运算以支持这些复杂的张量。

但是，有一个严重的commitment将此重要功能添加到pytorch中，这似乎是一个非常活跃的问题。