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

- old
+ new

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