aboutsummaryrefslogtreecommitdiff
path: root/cumatrix_example.lua
blob: f26d7f4274e86cdc346c60b1d4e68274b9fec7e9 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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)
print(fm:softmax())
print(dm)
print(dm:softmax())