Skip to content

Commit

Permalink
Clarify role of is_layer_type
Browse files Browse the repository at this point in the history
  • Loading branch information
APJansen committed Feb 16, 2024
1 parent 12315a1 commit 72a751b
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions n3fit/src/n3fit/backends/keras_backend/multi_dense.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 72a751b

Please sign in to comment.