flax.linen.Dense — Flax documentation

Union[None, str, jax._src.lax.lax.Precision, Tuple[str, str], Tuple[jax._src.lax.lax.Precision, jax._src.lax.lax.Precision]] initializer function for the weight matrix. Callable[[Any, Tuple[int, …], Any], Any] initializer function for the bias. Callable[[Any, Tuple[int, …], Any], Any] Applies a linear transformation to the inputs along the last dimension. inputs – The nd-array to be transformed. The transformed input. Methods Previous flax.linen.switch Next flax.linen.DenseGeneral By The Flax authors © Copyright 2020, The Flax authors.