search

williamFalcon / pytorch-complex-tensor

Unofficial complex tensor and scalar support for Pytorch
MIT License
78 stars 8 forks source link

Can it support to transfer CUDA torch.tensor to CUDA complex tensor？ #7

Open KellyGong opened 4 years ago

KellyGong commented 4 years ago

` import pytorch_complex_tensor as pct import torch

if name == 'main': C = torch.FloatTensor([[1, 1, 1], [2, 2, 2], [3, 3, 3], [4, 4, 4]]) C = C.cuda() print(pct.ComplexTensor(C)) `

the output is

Process finished with exit code -1073741819 (0xC0000005)

Thanks! Best wishes