neural_tangents.stax.FanInSum
- neural_tangents.stax.FanInSum()[source]
Layer construction function for a fan-in sum layer.
This layer takes a number of inputs (e.g. produced by FanOut) and sums the inputs to produce a single output.
- Return type
Tuple[InitFn,ApplyFn,LayerKernelFn,MaskFn]- Returns
(init_fn, apply_fn, kernel_fn).