jax.numpy.frompyfunc

目录

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