B×30
B×20
B×20
B×20
B×20
B×10
var_0
float32[B,30]
Gemm
float32[30,20]
B
〈30×20〉
float32[20]
C
〈20〉
BatchNormalization
batch_norm_0
float32[20]
scale
〈20〉
float32[20]
B
〈20〉
float32[20]
input_mean
〈20〉
float32[20]
input_var
〈20〉
Dropout
Dropout_0
float32
ratio
= 0.10000000…
boolean
training_mode
= true
Gelu
gelu_0
Gemm
float32[20,10]
B
〈20×10〉
float32[10]
C
〈10〉
gemm_out_1
float32[B,10]
×
jax_model
❮
Version
{version}
Copyright ©
Lutz Roeder
Open Model…
.
.
.
OK
≡