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

时间:2024-06-03 06:39:18
【文件属性】:

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

文件大小:15KB

文件格式:ZIP

更新时间:2024-06-03 06:39:18

Python

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 漂亮的印刷


【文件预览】:
pytorch-complex-tensor-master
----setup.py(479B)
----pytorch_complex_tensor()
--------complex_tensor.py(10KB)
--------complex_grad.py(2KB)
--------__init__.py(188B)
--------torch_complex.py(719B)
--------complex_scalar.py(538B)
----.gitignore(1KB)
----Complex demo example.ipynb(11KB)
----requirements.txt(319B)
----LICENSE(1KB)
----README.md(3KB)
----tests()
--------__init__.py(0B)
--------pytorch_complex_tensor_test.py(8KB)
--------torch_complex_test.py(956B)
----update.sh(276B)
----.circleci()
--------config.yml(2KB)

网友评论