import { Callout } from "nextra-theme-docs" import { Tab, Tabs } from "nextra-theme-docs"
Extracts the representation of CLS Token
ClassPool(0)
index
: Index of the CLS token to be extracted.
(x: jaxtyping.Float[Tensor, '... n_in d']) -> typing.Union