Start softmax with first logit

1 files changed, 1 insertions, 1 deletions

diff --git a/main.go b/main.go
index 6b5bd84..319a6b5 100644
--- a/main.go
+++ b/main.go
@@ -56,7 +56,7 @@ func main() {
 }
 
 func softmax(logits []float32) []float32 {
-	m := float32(0)
+	m := logits[0]
 
 	for _, v := range logits {
 		if v > m {