From 03417140e4f64533bac7e65566ca5936f54a992e Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Mon, 10 Nov 2025 19:29:12 +0100
Subject: Start softmax with first logit

---
 main.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 {
-- 
cgit v1.3.1