diff options
author | Dominik Grewe <dominikg@google.com> | 2015-07-10 15:41:20 +0300 |
---|---|---|
committer | Dominik Grewe <dominikg@google.com> | 2015-07-20 11:58:25 +0300 |
commit | f7df82a5a032442d4de77c2c0ed3e25a89667f1a (patch) | |
tree | 1d6de591ee1dfc5f7a20dbce750301b135087f30 /TensorMath.lua | |
parent | 90f0cde041b645ac4b3dd3179a3d1af7e815a994 (diff) |
Element-wise min and max operations.
Add cmin & cmax operation for element-wise min and max between two
tensors and between a tensor and a scalar.
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 1fba493..a025c68 100644 --- a/TensorMath.lua +++ b/TensorMath.lua @@ -457,6 +457,18 @@ for _,Tensor in ipairs({"ByteTensor", "CharTensor", {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("trace", cname("trace"), {{name=Tensor}, |