sympytorch:将SymPy表达式转换为PyTorch模块

时间:2024-03-23 21:34:20
【文件属性】:

文件名称:sympytorch:将SymPy表达式转换为PyTorch模块

文件大小:9KB

文件格式:ZIP

更新时间:2024-03-23 21:34:20

Python

象征火炬 一个微库,方便将SymPy表达式转换为PyTorch模块。 所有SymPy浮点数都成为可训练的参数。 所有SymPy符号都是模块的输入。 安装 pip install git+https://github.com/patrick-kidger/sympytorch.git 例子 import sympy , torch , sympytorch x = sympy . symbols ( 'x_name' ) cosx = 1.0 * sympy . cos ( x ) sinx = 2.0 * sympy . sin ( x ) mod = sympytorch . SymPyModule ( expressions = [ cosx , sinx ]) x_ = torch . rand ( 3 ) out = mod ( x_name = x_ ) # out has


【文件预览】:
sympytorch-master
----LICENSE(11KB)
----setup.py(2KB)
----README.md(1KB)
----sympytorch()
--------__init__.py(62B)
--------sympy_module.py(4KB)
----tests()
--------test_some.py(1KB)
----.gitignore(27B)

网友评论