B×28×28×1B×1×28×28B×32×28×28B×28×28×32B×32×14×14B×64×14×14B×14×14×64B×64×7×7B×7×7×64B×7×7×64412B×3136B×256B×256B×10var_0float32[B,28,28,1]Transposetranspose_pre_0Convconv_0float32[32,1,3,3]W〈32×1×3×3〉float32[32]B〈32〉Relurelu_0AveragePoolavg_pool_0Convconv_1float32[64,32,3,3]W〈64×32×3×3〉float32[64]B〈64〉Relurelu_1AveragePoolavg_pool_1Transposetranspose_post_3ShapeGatherint64indices = 0Unsqueezeint64[1]axes〈1〉Concat〈…〉Reshapereshape_0Gemmfloat32[3136,256]B〈3136×256〉float32[256]C〈256〉Relurelu_2Gemmfloat32[256,10]B〈256×10〉float32[10]C〈10〉gemm_out_1float32[B,10]