diff options
author | Pavan Yalamanchili <pyalamanchili@twitter.com> | 2016-12-26 07:21:22 +0300 |
---|---|---|
committer | Pavan Yalamanchili <pyalamanchili@twitter.com> | 2016-12-28 00:37:42 +0300 |
commit | 12ffddf5cd7e29944246497ad7a499169c4bbd4e (patch) | |
tree | 119b8540ec9b1afdeb8eb32dce7a15a61d1db7c3 /TensorMath.lua | |
parent | 33449f817dbd5f32cb0162a4f500b6753152718d (diff) |
Adding support for empty tensors in cat, catArray
Diffstat (limited to 'TensorMath.lua')
-rw-r--r-- | TensorMath.lua | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/TensorMath.lua b/TensorMath.lua index 3072ea2..e16748e 100644 --- a/TensorMath.lua +++ b/TensorMath.lua @@ -906,11 +906,11 @@ for k, Tensor_ in pairs(handledTypenames) do {{name=Tensor, default=true, returned=true}, {name=Tensor}, {name=Tensor}, - {name="index", default=lastdim(2)}}, + {name="index", default=-1}}, cname("catArray"), {{name=Tensor, default=true, returned=true}, {name=Tensor .. "Array"}, - {name="index", default=lastdimarray(2)}}) + {name="index", default=-1}}) wrap("geometric", cname("geometric"), @@ -1864,11 +1864,11 @@ wrap("cat", {{name=Tensor, default=true, returned=true}, {name=Tensor}, {name=Tensor}, - {name="index", default=lastdim(2)}}, + {name="index", default=-1}}, cname("catArray"), {{name=Tensor, default=true, returned=true}, {name=Tensor .. "Array"}, - {name="index", default=lastdimarray(2)}}) + {name="index", default=-1}}) wrap("nonzero", cname("nonzero"), |