pytorch-complex-tensor:Pytorch的非官方复张量和标量支持

pytorch复合张量 非官方的复杂Tensor支持Pytorch 这个怎么运作 将张量的前半部分视作实数，将第二个视作虚数。 实现了一些算术运算以模拟复杂的算术。 支持渐变。 安装 pip install pytorch-complex-tensor 例子： 容易导入 from pytorch_complex_tensor import ComplexTensor 初始张量 # equivalent to: # np.asarray([[1+3j, 1+3j, 1+3j], [2+4j, 2+4j, 2+4j]]).astype(np.complex64) C = ComplexTensor ([[ 1 , 1 , 1 ], [ 2 , 2 , 2 ], [ 3 , 3 , 3 ], [ 4 , 4 , 4 ]]) C . requires_grad = True 漂亮的印刷