Skip to content

Commit

Permalink
update docs
Browse files Browse the repository at this point in the history
  • Loading branch information
pfackeldey committed Mar 20, 2024
1 parent a2b27bf commit ca9651d
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/evermore/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,11 +53,11 @@ def sum_leaves(tree: PyTree) -> Array:

def tree_stack(trees: list[PyTree], broadcast_leaves: bool = False) -> PyTree:
"""
Turn an array of `evm.Modifier`(s) into a `evm.Modifier` of arrays.
Turn an array of evm.Modifier(s) into a evm.Modifier of arrays.
It is important that the `jax.Array`(s) of the underlying `evm.Parameter` have the same shape.
Same applies for the effect leaves (e.g. `width`). However, the effect leaves can be
broadcasted to the same shape if `broadcast_effect_leaves` is set to `True`.
It is important that the jax.Array(s) of the underlying evm.Parameter have the same shape.
Same applies for the effect leaves (e.g. width). However, the effect leaves can be
broadcasted to the same shape if broadcast_effect_leaves is set to True.
Example:
Expand Down

0 comments on commit ca9651d

Please sign in to comment.