B×2B×2var_0float64[B,2]Softmaxsoftmax_0var_1float64[B,2]