gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-03 23:35:15 24 4
gpt4 key购买 nike

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

最佳答案

PyTorch 确实有 complex number support .

试试这个:

import torch

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

关于pytorch - 如何将张量转换为 Pytorch 中的复杂类型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59717519/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com