jax.tree_util.tree_reduce

jax.tree_util.tree_reduce(function, tree, initializer=<object object>, is_leaf=None)

别名 jax.tree.reduce()。

参数:

• function (Callable[[T, Any], T])

• tree (Any)

• initializer (Any)

• is_leaf (Callable[[Any], bool] | None)

返回类型:

T