m = 10 n = 10 fm = nerv.FloatCuMatrix(m, n) dm = nerv.DoubleCuMatrix(m, n) for i = 0, m - 1 do for j = 0, n - 1 do -- local t = math.random(10) t = i / (j + 1) fm[i][j] = t dm[i][j] = t end end print(fm) fs = fm:softmax() print(fs) print(dm) ds = dm:softmax() print(ds) print(fs + fs) print(fs - fs)