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