aboutsummaryrefslogtreecommitdiff
path: root/examples/cumatrix_example.lua
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-05-25 16:07:34 +0800
committerDeterminant <[email protected]>2015-05-25 16:07:34 +0800
commitbe922fa355fe385b45a2720d48eaf0c809e1874a (patch)
tree6ec8747f0463ffb36445d7cadec567c00b8fdbdf /examples/cumatrix_example.lua
parent3036cc0f7597d72a36b1923ab5e0c56a4490be31 (diff)
move example files to a dedicated directory
Diffstat (limited to 'examples/cumatrix_example.lua')
-rw-r--r--examples/cumatrix_example.lua23
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)