【文件属性】:
文件名称:pytorch-complex-tensor:Pytorch的非官方复张量和标量支持
文件大小:15KB
文件格式:ZIP
更新时间:2021-05-14 12:52:38
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)