1 files changed, 3 insertions, 3 deletions
diff --git a/init.lua b/init.lua
index 4cd88ea..c3cdbf4 100644
--- a/init.lua
+++ b/init.lua
@@ -6,11 +6,11 @@ qt.QImage.fromTensor = function(tensor, scale)
                           return tensor.qttorch.QImageFromTensor(tensor, scale)
                        end
 
-qt.QImage.toTensor = function(self, tensor, scale)
+qt.QImage.toTensor = function(self, tensor, scale, depth)
                         if type(tensor) == 'userdata' then
-                           return tensor.qttorch.QImageToTensor(self, tensor, scale)
+                           return tensor.qttorch.QImageToTensor(self, tensor, scale, depth)
                         else
                            local t = torch.getmetatable(torch.getdefaulttensortype())
-                           return t.qttorch.QImageToTensor(self, tensor, scale)
+                           return t.qttorch.QImageToTensor(self, tensor, scale, depth)
                         end
                      end