我想用Pytorch进行一些量子力学计算,其中的量有时很复杂。我想知道如何将现有的实际张量转换为复杂类型。
PyTorch确实支持复数。
尝试这个:
import torch a = torch.tensor([1.0, 2.0], dtype=torch.double) b = a.type(torch.complex64)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
点击生成二维码
我来说两句