diff options
author | soumith <soumith@fb.com> | 2017-07-17 18:56:18 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2017-07-17 18:56:18 +0300 |
commit | caf84f3af0b1e9f6bd5c4129c01345f7bd72e431 (patch) | |
tree | 7f57599f73796e46b4799525531c3f0b4b0ba13d | |
parent | f1cad197b012c09490ab07d5880282f58ec1cb53 (diff) |
fix cwrap
-rw-r--r-- | TensorMath.lua | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/TensorMath.lua b/TensorMath.lua index d45db3e..ffd96e2 100644 --- a/TensorMath.lua +++ b/TensorMath.lua @@ -1057,7 +1057,7 @@ for k, Tensor_ in pairs(handledTypenames) do {name='CudaLongTensor', default=true, returned=true, method={default='nil'}}, {name=Tensor, default=true, returned=true, method={default='nil'}}}) - + wrap("multinomialAlias_", cname("multinomialAliasDraw"), {{name="CudaLongTensor", default=true, returned=true, method={default='nil'}}, @@ -1961,7 +1961,7 @@ wrap("multinomialAlias_", {name="CudaLongTensor"}, {name=Tensor} }) - + wrap("clamp", cname("clamp"), {{name=Tensor, default=true, returned=true, method={default='nil'}}, @@ -2166,6 +2166,7 @@ for _,name in ipairs({"var", "std"}) do wrap(name, cname(name .. "all"), {{name=Tensor}, + {name="boolean", default=false}, {name=real, creturned=true}}, cname(name), {{name=Tensor, default=true, returned=true}, |