Function: poolAndNormalize()

function poolAndNormalize(
  tokenStates: TokenStates3D,
  pooling: string,
  normalize: boolean,
): Pooled2D;

Defined in: src/batteries/embeddings/transformers_js/pooling.ts:88

Pool raw [batch, seq, hidden] token states into [batch, hidden] vectors and optionally L2-normalize — the deterministic battery-owned path (poolingOwner: 'battery').

Parameters

Parameter	Type	Description
tokenStates	TokenStates3D	The [batch, seq, hidden] raw states from pooling:'none'.
pooling	string	The pooling strategy ('mean' default).
normalize	boolean	Whether to L2-normalize each pooled vector (default true).

Returns

Pooled2D

One pooled (and optionally normalized) vector per batch row.