From ab12a9583bdd39884fde9bc2444e6fd1bc5f518e Mon Sep 17 00:00:00 2001 From: Determinant Date: Sun, 31 May 2015 11:18:16 +0800 Subject: add async copy by index; add MMatrixInt --- matrix/init.lua | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) (limited to 'matrix/init.lua') diff --git a/matrix/init.lua b/matrix/init.lua index 0075668..057b085 100644 --- a/matrix/init.lua +++ b/matrix/init.lua @@ -2,17 +2,22 @@ function nerv.Matrix:__tostring__() local ncol = self:ncol() local nrow = self:nrow() local strt = {} - + local fmt + if self.fmt then + fmt = self.fmt + else + fmt = "%.10f " + end if nrow == 1 then for col = 0, ncol - 1 do - table.insert(strt, string.format("%f ", self[col])) + table.insert(strt, string.format(fmt, self[col])) end table.insert(strt, "\n") else for row = 0, nrow - 1 do local rp = self[row] for col = 0, ncol - 1 do - table.insert(strt, string.format("%.10f ", rp[col])) + table.insert(strt, string.format(fmt, rp[col])) end table.insert(strt, "\n") end @@ -21,6 +26,8 @@ function nerv.Matrix:__tostring__() return table.concat(strt) end +nerv.MMatrixInt.fmt = "%d " + function nerv.CuMatrix:__add__(b) c = self:create() c:add(self, b, 1.0, 1.0) -- cgit v1.2.3