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