From ef8e70d0dbdfb2a2263201d38bcf66f0febee658 Mon Sep 17 00:00:00 2001 From: Nicholas Leonard Date: Thu, 12 Feb 2015 14:02:12 -0500 Subject: [PATCH] extend.lua --- extend.lua | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 extend.lua diff --git a/extend.lua b/extend.lua new file mode 100644 index 0000000..9e8de8c --- /dev/null +++ b/extend.lua @@ -0,0 +1,10 @@ + + +function torchx.extend(types, funcs, alias) + for _,type in ipairs(types) do + local metatable = torch.getmetatable(alias and ('torch.' .. type .. 'Tensor') or type) + for funcname, func in pairs(funcs) do + rawset(metatable, funcname, func) + end + end +end