@ptrblck  why the results of softmax are in two columns?
like that :
softmax de test [array([0.35895655, 0.6410435 ], dtype=float32), array([0.5, 0.5], dtype=float32), array([0.38063172, 0.61936826], dtype=float32), array([0.30245292, 0.6975471 ], dtype=float32), array([0.00381802, 0.99618196], dtype=float32),…, array([0.5064734 , 0.49352652], dtype=float32).