Module layers

Source

Structsยง

Dense
Feed-forward function with activiation function.
DenseConfig
Configuration for Dense layer.
Pooling
Performs pooling (max or mean) on the token embeddings.
PoolingConfig
Configuration for Pooling layer.