3×28×28×1
3×10
var_0
float32[3,28,28,1]
deterministic
boolean
VisionTransformer_1
VisionTransformer_1
ƒ
Show Function Definition
var_498
float32[3,10]
×
vit_conv_embedding
VisionTransformer_1
ConvEmbedding_2
ConcatClsToken_2
PositionalEmbedding_2
TransformerStack_2
TransformerBlock_2
MultiHeadAttention_2
attention_0
FeedForward_2
TransformerBlock_5
MultiHeadAttention_5
attention_1
FeedForward_5
TransformerBlock_8
MultiHeadAttention_8
attention_2
FeedForward_8
TransformerBlock_11
MultiHeadAttention_11
attention_3
FeedForward_11
TransformerBlock_14
MultiHeadAttention_14
attention_4
FeedForward_14
TransformerBlock_17
MultiHeadAttention_17
attention_5
FeedForward_17
ClassificationHead_2
get_first_token_0
❮
Version
{version}
Copyright ©
Lutz Roeder
Open Model…
.
.
.
OK
≡