lib/torch/utils/data/tensor_dataset.rb in torch-rb-0.3.5 vs lib/torch/utils/data/tensor_dataset.rb in torch-rb-0.3.6

@@ -14,9 +14,11 @@
         end
 
         def size
           @tensors[0].size(0)
         end
+        alias_method :length, :size
+        alias_method :count, :size
       end
     end
   end
 end