Как я могу привести тензор к сложному типу в Pytorch?

Question

Я хочу сделать некоторые вычисления квантовой механики с Pytorch, где величины иногда бывают сложными. Я хотел бы знать, как я могу привести существующий реальный тензор к сложному типу.

Answer 1

PyTorch имеет поддержку комплексных чисел .

Попробуйте:

import torch

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

Answer 2

AFAIK pytorch не поддерживает сложные тензоры.
Вы можете представить сложный тензор, используя два действительных. Однако для поддержки этих сложных тензоров вам придется реализовать все соответствующие операции basi c arithmeti c.

Однако , существует серьезное обязательство добавить эту важную функцию в pytorch, и это кажется очень активной проблемой.