地址：
from jax import make_jaxpr
def f(x, y):
return jnp.dot(x + 1, y + 1)
make_jaxpr(f)(x, y)
Jax的加速层的伪代码/中间层代码的生成和查看
发布时间 2024-01-09 12:34:04作者: Angry_Panda
地址：
from jax import make_jaxpr
def f(x, y):
return jnp.dot(x + 1, y + 1)
make_jaxpr(f)(x, y)