jax.tree_util.tree_reduce#
- jax.tree_util.tree_reduce(function, tree, initializer=<object object>, is_leaf=None)[source]#
jax.tree.reduce()的别名。- 参数：
function (Callable[[T, Any], T])
tree (Any)
initializer (Any)
is_leaf (Callable[[Any], bool] | None | None)
- 返回类型：
T