Skip to content
1 min read · 96 words

Variable: defaultPoolAndNormalize

ts
const defaultPoolAndNormalize: (
  tokenStates: TokenStates3D,
  pooling: string,
  normalize: boolean,
) => Pooled2D = poolAndNormalize;

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

Default poolAndNormalize.

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

Parameters

ParameterTypeDescription
tokenStatesTokenStates3DThe [batch, seq, hidden] raw states from pooling:'none'.
poolingstringThe pooling strategy ('mean' default).
normalizebooleanWhether to L2-normalize each pooled vector (default true).

Returns

Pooled2D

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