ClassPool

Extracts the representation of CLS Token

ClassPool(0)

Parameters

• index: Index of the CLS token to be extracted.

Forward

(x: jaxtyping.Float[Tensor, '... n_in d']) -> typing.Union