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