public
ClassTokens
— parametric type
ClassTokens(dim; init = Flux.zeros32)
Appends class tokens to an input with embedding dimension dim
for use in many vision transformer models.
public
ClassTokens
— parametric type
ClassTokens(dim; init = Flux.zeros32)
Appends class tokens to an input with embedding dimension dim
for use in many vision transformer models.