diff options
author | Dominik Grewe <dominikg@google.com> | 2015-07-10 17:02:02 +0300 |
---|---|---|
committer | Dominik Grewe <dominikg@google.com> | 2015-07-10 17:02:02 +0300 |
commit | c0730eb39c3a84013e60d907265fc36ab53fb113 (patch) | |
tree | 4c8eae439a5505dcffc8936fada2d62755380ba6 /TensorMath.lua | |
parent | 318fb5b38a189a7c27f016f22145bc42a09ff188 (diff) |
Implementations of component-wise min and max.
Diffstat (limited to 'TensorMath.lua')
-rw-r--r-- | TensorMath.lua | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/TensorMath.lua b/TensorMath.lua index 466164a..4f86d55 100644 --- a/TensorMath.lua +++ b/TensorMath.lua @@ -549,6 +549,18 @@ for _,name in ipairs({"min", "max"}) do {name="index"}}) end +for _,name in ipairs({"cmin", "cmax"}) do + wrap(name, + cname(name), + {{name=Tensor, default=true, returned=true}, + {name=Tensor, method={default=1}}, + {name=Tensor}}, + cname(name .. "Value"), + {{name=Tensor, default=true, returned=true}, + {name=Tensor, method={default=1}}, + {name=real}}) +end + wrap("tril", cname("tril"), {{name=Tensor, default=true, returned=true}, |