tensor.rb in torch-rb-0.2.1

- old
+ new

@@ -35,9 +35,13 @@
 
     def cpu
       to("cpu")
     end
 
+    def cuda
+      to("cuda")
+    end
+
     def size(dim = nil)
       if dim
         _size_int(dim)
       else
         shape