jax.numpy.frompyfunc# jax.numpy.frompyfunc(func, /, nin, nout, *, identity=None)[源代码][源代码]# 从任意兼容JAX的标量函数创建一个JAX ufunc。 参数: func (Callable[..., Any]) – 一个可调用对象,接受 nin 个标量参数并返回 nout 个输出。 nin (int) – 指定标量输入数量的整数 nout (int) – 指定标量输出数量的整数 identity (Any) – (可选) 一个标量,指定操作的身份(如果有)。 返回: jax.numpy.ufunc 的 func 包装器。 返回类型: wrapped