tree_concatenate

bayesflow.utils.tree_concatenate(structures: Sequence[T], axis: int = 0, numpy: bool = None) → T

Concatenate all tensors in the given sequence of nested structures. All objects in the given sequence must have the same structure. The output will adhere to this structure.

Parameters:

• structures – A sequence of nested structures of tensors. All structures in the sequence must have the same layout. Tensors in the same layout location must have compatible shapes for concatenation.

• axis – The axis along which to concatenate tensors.

• numpy – Whether to use numpy or keras for concatenation. Will convert all items in the structures to numpy arrays if True, tensors otherwise. Defaults to True if all tensors are numpy arrays, False otherwise.

Returns:

A structure of concatenated tensors with the same layout as each input structure.