split_tensors#

bayesflow.utils.split_tensors(data: Mapping[any, Tensor], axis: int = -1) Mapping[any, Tensor][source]#

Split tensors in the dictionary along the given axis.