def rows.softmax

bend-ml-tensor@0.1.2.0 · main.bend:347

import bend-ml-tensor@0.1.2.0/main.bend as BendMlTensor

import Base

347def rows.softmax(xs: List<&2, List<&2, F32>>) -> List<&2, List<&2, F32>>:
348  match xs:
349    case Nil{}:
350      Nil{}
351    case Con{r, t}:
352      softmax.l(r) <> rows.softmax(t)

Uses (3)

Used by (1)

bend-ml-tensor@0.1.2.0/main:Mat.softmax