diff options
Diffstat (limited to 'nerv/examples/cumatrix_example.lua')
-rw-r--r-- | nerv/examples/cumatrix_example.lua | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/nerv/examples/cumatrix_example.lua b/nerv/examples/cumatrix_example.lua new file mode 100644 index 0000000..544fc7f --- /dev/null +++ b/nerv/examples/cumatrix_example.lua @@ -0,0 +1,31 @@ +m = 4 +n = 4 +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:create() +fs:softmax(fm) +-- print(fs) +print(dm) +ds = dm:create() +ds:softmax(dm) +-- print(ds) +print(fs) +print(fs + fs) +print(ds + ds) +print(fs - fs) +print(ds - ds) + +a = fs:create() +a:mul_elem(fs, fs) +print(a) +a:log_elem(fs) +print(a) |