From 8f18aaba322cf717b8805ebfcacb5a3108d3dda8 Mon Sep 17 00:00:00 2001 From: Determinant Date: Fri, 22 May 2015 10:26:12 +0800 Subject: add base class CuMatrix for CuMatrixFloat and CuMatrixDouble --- cumatrix_example.lua | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'cumatrix_example.lua') diff --git a/cumatrix_example.lua b/cumatrix_example.lua index aeb5828..26e1dc4 100644 --- a/cumatrix_example.lua +++ b/cumatrix_example.lua @@ -1,7 +1,7 @@ m = 10 n = 10 -fm = nerv.FloatCuMatrix(m, n) -dm = nerv.DoubleCuMatrix(m, n) +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) @@ -10,11 +10,11 @@ for i = 0, m - 1 do dm[i][j] = t end end -print(fm) +-- print(fm) fs = fm:softmax() -print(fs) -print(dm) +-- print(fs) +-- print(dm) ds = dm:softmax() -print(ds) +-- print(ds) print(fs + fs) print(fs - fs) -- cgit v1.2.3