You can find the actual computation for the output shape in the docs 11.



 blackbirdbarber:

``
I see now that bs has key role, but I wonder why only w is taken in account.
Why b part is ignored?


I’m a bit confused. Are you looking for the number of parameters (weight + bias) or the output activation shape?