diff options
author | Determinant <[email protected]> | 2015-05-25 17:27:29 +0800 |
---|---|---|
committer | Determinant <[email protected]> | 2015-05-25 17:27:29 +0800 |
commit | 81c115e09f9449ae61c7352edd77c68e9029f2dc (patch) | |
tree | 284196680a794d4317d667e6a6e04fa5b3ee8c75 /examples | |
parent | c1e714052f0c2654ebe2a92d6961382146450b9e (diff) |
add "copy from/to cumatrix" functionality
Diffstat (limited to 'examples')
-rw-r--r-- | examples/cumatrix_from_mmatrix.lua | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/examples/cumatrix_from_mmatrix.lua b/examples/cumatrix_from_mmatrix.lua new file mode 100644 index 0000000..fba8a90 --- /dev/null +++ b/examples/cumatrix_from_mmatrix.lua @@ -0,0 +1,23 @@ +m = 10 +n = 10 +fm = nerv.MMatrixFloat(m, n) +dm = nerv.MMatrixDouble(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) +print(dm) + +fc = nerv.CuMatrixFloat(m, n) +dc = nerv.CuMatrixDouble(m, n) +fc:copy_from(fm) +dc:copy_from(dm) +print(fc) +print(dc) +print(fc:softmax()) +print(dc:softmax()) |