From 72a751bb68e7d9148b59bca4268a9a74ad877fee Mon Sep 17 00:00:00 2001 From: Aron Date: Fri, 16 Feb 2024 10:11:47 +0100 Subject: [PATCH] Clarify role of is_layer_type --- n3fit/src/n3fit/backends/keras_backend/multi_dense.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/n3fit/src/n3fit/backends/keras_backend/multi_dense.py b/n3fit/src/n3fit/backends/keras_backend/multi_dense.py index e2bd9d471e..b8caa9a3fd 100644 --- a/n3fit/src/n3fit/backends/keras_backend/multi_dense.py +++ b/n3fit/src/n3fit/backends/keras_backend/multi_dense.py @@ -9,8 +9,9 @@ class MultiDense(Dense): """ Dense layer for multiple replicas at the same time. - For the first layer in the network, the input shape is (batch_size, gridsize, features), - still without a replica axis. In this case this layer acts as a stack of single dense layers, + For the first layer in the network, (for which ``is_first_layer`` should be set to True), + the input shape is (batch_size, gridsize, features), still without a replica axis. + In this case this layer acts as a stack of single dense layers, with their own kernel and bias, acting on the same input. For subsequent layers, the input already contains multiple replicas, and the shape is