辛普森公司

2022-05-05

以下示例是关于Python中包含辛普森公司用法的示例代码,想了解辛普森公司的具体用法?辛普森公司怎么用?辛普森公司使用的例子?那么可以参考以下相关源代码片段来学习它的具体使用方法。

用sympy的`lambdify'函数生成numpy函数。

文件名:lambdify_sympy.py[英]:Sympy lambdify源码类型:Python
import numpy as np
import sympy
import jax
import jax.numpy as jnp
from sympy.functions.special.polynomials import hermite


x = sympy.symbols("x")  # define symbol 
H_3 = hermite(3, x)  # 3th Hermite polynomial
eval_hermite = sympy.lambdify(x, H_3, "numpy") # args: symbol, expression, backend

# Evaluate polynomial
x_vals = np.linspace(-1, 1, 100)
y_vals = eval_hermite(x_vals)

# JAX transforms are also available

eval_hermite = sympy.lambdify(x, H_3, modules=[jnp])
y_vals = jax.jit(eval_hermite)(x_vals)

本文地址:https://www.itbaoku.cn/snippets/785109.html