象征火炬 一个微库,方便将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