From b2cd6fbf03c3e0a17602d87b832309a24cf2b458 Mon Sep 17 00:00:00 2001 From: Aron Date: Mon, 4 Mar 2024 11:21:24 +0100 Subject: [PATCH] list->tuple in output shape --- n3fit/src/n3fit/backends/keras_backend/multi_dense.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/n3fit/src/n3fit/backends/keras_backend/multi_dense.py b/n3fit/src/n3fit/backends/keras_backend/multi_dense.py index d0c0bdaf4a..9fa0ac5835 100644 --- a/n3fit/src/n3fit/backends/keras_backend/multi_dense.py +++ b/n3fit/src/n3fit/backends/keras_backend/multi_dense.py @@ -125,7 +125,7 @@ def compute_output_shape(self, input_shape): output_shape = super().compute_output_shape(input_shape) # Add back the replica axis to the output shape. - output_shape = output_shape[:1] + [self.replicas] + output_shape[1:] + output_shape = output_shape[:1] + (self.replicas,) + output_shape[1:] return output_shape