jax.numpy.fromfunction

jax.numpy.fromfunction#

jax.numpy.fromfunction(function, shape, *, dtype=<class 'float'>, **kwargs)[源代码][源代码]#

通过在每个坐标上执行一个函数来构造一个数组。

LAX-backend 实现的 numpy.fromfunction()。

原始文档字符串如下。

因此，生成的数组在坐标 (x, y, z) 处具有值 fn(x, y, z)。

参数:

  • function (callable) – 该函数使用 N 个参数调用，其中 N 是 shape 的秩。每个参数表示沿着特定轴变化的数组坐标。例如，如果 shape 是 (2, 2)，那么参数将是 array([[0, 0], [1, 1]]) 和 array([[0, 1], [0, 1]])。

  • shape ((N,) tuple of ints) – 输出数组的形状，这也决定了传递给 function 的坐标数组的形状。

  • dtype (data-type, optional) – 传递给 function 的坐标数组的 数据类型。默认情况下，dtype 是浮点型。

返回:

fromfunction – 调用 function 的结果直接返回。因此，fromfunction 的形状完全由 function 决定。如果 function 返回一个标量值，fromfunction 的形状将不会匹配 shape 参数。

返回类型:

any

目录