Added CudaTensor.expand and CudaTensor.expandAs

1 files changed, 7 insertions, 0 deletions

diff --git a/Tensor.lua b/Tensor.lua
index fc4f283..c8e74e8 100644
--- a/Tensor.lua
+++ b/Tensor.lua
@@ -53,3 +53,10 @@ for _,func in ipairs({'addmv',
                                end
                             end      
 end
+
+do
+    local metatable = torch.getmetatable('torch.CudaTensor')
+    for _,func in pairs{'expand', 'expandAs'} do
+        rawset(metatable, func, torch[func])
+    end
+end