extend.lua

nicholas-leonard · Feb 12, 2015 · ef8e70d · ef8e70d
1 parent 1db2eff
commit ef8e70d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/extend.lua 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