diff options
author | Determinant <[email protected]> | 2015-05-28 14:31:31 +0800 |
---|---|---|
committer | Determinant <[email protected]> | 2015-05-28 14:31:31 +0800 |
commit | c13115662e739b434f1071eb623a41a39d8b4985 (patch) | |
tree | 9368b5706ef9ddb3002369d3193f525ad9814201 /matrix/init.lua | |
parent | 382106f36025f76e2f5d04c44b9ccb0998cf40cf (diff) |
should support multiple input/output for layers
Diffstat (limited to 'matrix/init.lua')
-rw-r--r-- | matrix/init.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/matrix/init.lua b/matrix/init.lua index a04e83a..0075668 100644 --- a/matrix/init.lua +++ b/matrix/init.lua @@ -41,12 +41,12 @@ end function nerv.CuMatrixFloat.new_from_host(mat) local res = nerv.CuMatrixFloat(mat:nrow(), mat:ncol()) - res:copy_from(mat) + res:copy_fromh(mat) return res end function nerv.CuMatrixFloat:new_to_host() local res = nerv.MMatrixFloat(self:nrow(), self:ncol()) - self:copy_to(res) + self:copy_toh(res) return res end |