add support for equal in cutorch

1 files changed, 12 insertions, 0 deletions

diff --git a/TensorMath.lua b/TensorMath.lua
index f47c6cc..d092dc2 100644
--- a/TensorMath.lua
+++ b/TensorMath.lua
@@ -673,6 +673,12 @@ for k, Tensor_ in pairs(handledTypenames) do
             {name=Tensor, method={default=1}},
             {name=real}})
 
+    wrap("equal",
+         cname("equal"),
+         {{name=Tensor},
+          {name=Tensor},
+          {name="boolean", creturned=true}})
+
     for _, name in ipairs({"cmul", "cpow", "cdiv"}) do
        wrap(name,
             cname(name),
@@ -1330,6 +1336,12 @@ wrap("remainder",
         {name=Tensor, method={default=1}},
         {name=real}})
 
+wrap("equal",
+     cname("equal"),
+     {{name=Tensor},
+      {name=Tensor},
+      {name="boolean", creturned=true}})
+
 for _, name in ipairs({"cmul", "cpow", "cdiv"}) do
   wrap(name,
        cname(name),