find returns a table

README.md

@@ -30,16 +30,14 @@ Example:
 
 <a name='torch.find'/>
 ### [res] torch.find(tensor, val) ###
-Finds all indices of a given value `val` in Tensor `tensor`. Returns a `torch.LongTensor` of these indices.
+Finds all indices of a given value `val` in Tensor `tensor`. 
+Returns a table of these indices.
 
 Example:
 ```lua
 > res = torch.find(torch.Tensor{1,2,3,1,1,2}, 1)
-> print(res)
- 1
- 4
- 5
-[torch.LongTensor of dimension 3]
+> print(unpack(res))
+ 1, 4, 5
 ```
 
 <a name='torch.group'/>

find.lua

@@ -1,6 +1,6 @@
 
 
-function torch.find(res, tensor, val, asTable)
+function torch.find(tensor, val)
    if not torch.isTensor(tensor) then
       asTable = val
       val = tensor
@@ -9,7 +9,6 @@ function torch.find(res, tensor, val, asTable)
          res = torch.LongTensor()
       end
    end
-   assert(tensor:dim() == 1, "torch.find only supports 1D tensors (for now)")
    local i = 1
    local indice = {}
    tensor:apply(function(x)
@@ -18,16 +17,6 @@ function torch.find(res, tensor, val, asTable)
          end
          i = i + 1
       end)
-   if asTable then
-      return indice
-   end
-   res:resize(#indice)
-   i = 0
-   res:apply(function()
-         i = i + 1
-         return indice[i]
-      end)
-   return res
+   return indice
 end
 
-torchx.Tensor.find = torch.find