aboutsummaryrefslogblamecommitdiff
path: root/cumatrix_example.lua
blob: 26e1dc4ad27bfb4adc88ac60f8c32f8f3d213e04 (plain) (tree)
1
2
3
4
5
6
7
8
9
10

      

                              

                       



                                    

       
            
                 

            
                 
            

              
m = 10
n = 10
fm = nerv.CuMatrixFloat(m, n)
dm = nerv.CuMatrixDouble(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)