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 --- matrix/init.lua | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'matrix/init.lua') diff --git a/matrix/init.lua b/matrix/init.lua index d6aab73..c33cf7e 100644 --- a/matrix/init.lua +++ b/matrix/init.lua @@ -20,3 +20,15 @@ function nerv.Matrix:__tostring__() table.insert(strt, string.format("[Matrix %d x %d]", nrow, ncol)) return table.concat(strt) end + +function nerv.CuMatrix:__add__(b) + c = self:create() + c:add(self, b, 1.0, 1.0) + return c +end + +function nerv.CuMatrix:__sub__(b) + c = self:create() + c:add(self, b, 1.0, -1.0) + return c +end -- cgit v1.2.3