diff options
author | Determinant <[email protected]> | 2015-05-25 16:07:34 +0800 |
---|---|---|
committer | Determinant <[email protected]> | 2015-05-25 16:07:34 +0800 |
commit | be922fa355fe385b45a2720d48eaf0c809e1874a (patch) | |
tree | 6ec8747f0463ffb36445d7cadec567c00b8fdbdf /examples/cumatrix_example.lua | |
parent | 3036cc0f7597d72a36b1923ab5e0c56a4490be31 (diff) |
move example files to a dedicated directory
Diffstat (limited to 'examples/cumatrix_example.lua')
-rw-r--r-- | examples/cumatrix_example.lua | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/examples/cumatrix_example.lua b/examples/cumatrix_example.lua new file mode 100644 index 0000000..f8235eb --- /dev/null +++ b/examples/cumatrix_example.lua @@ -0,0 +1,23 @@ +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) +print(fs + fs) +print(ds + ds) +print(fs - fs) +print(ds - ds) |