From 42d20ab23802ebf40aecf2780525af96df3c5f32 Mon Sep 17 00:00:00 2001 From: stefanradev93 Date: Wed, 12 Jun 2024 04:37:29 -0400 Subject: [PATCH] Bugfix concatenation axis --- bayesflow/experimental/utils/dictutils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bayesflow/experimental/utils/dictutils.py b/bayesflow/experimental/utils/dictutils.py index adee00fb3..c8d547117 100644 --- a/bayesflow/experimental/utils/dictutils.py +++ b/bayesflow/experimental/utils/dictutils.py @@ -19,7 +19,7 @@ def concatenate_tensors(tensor_dict: dict[str, Tensor], filter_list: list, axis: An optional axis can be specified (default: last axis). """ - return ops.concatenate([v for k, v in tensor_dict.items() if k in filter_list]) + return ops.concatenate([v for k, v in tensor_dict.items() if k in filter_list], axis=axis) def keras_kwargs(kwargs: dict):