i see. Thank you ! Is there a special name for “using a dense like this without a softmax”?
Maybe i can read more about it to make it clear. I saw something  similar to it called “multi parameter regression” but not sure