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)