torch-rb 0.3.2 → 0.3.7

data/lib/torch/utils/data/data_loader.rb

@@ -45,6 +45,8 @@ module Torch
         def size
           (@dataset.size / @batch_size.to_f).ceil
         end
+        alias_method :length, :size
+        alias_method :count, :size
 
         private
 

data/lib/torch/utils/data/tensor_dataset.rb

@@ -16,6 +16,8 @@ module Torch
         def size
           @tensors[0].size(0)
         end
+        alias_method :length, :size
+        alias_method :count, :size
       end
     end
   end

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.3.2"
+  VERSION = "0.3.7"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.3.7
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-08-24 00:00:00.000000000 Z
+date: 2020-09-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -238,6 +238,7 @@ files:
 - lib/torch/nn/tanhshrink.rb
 - lib/torch/nn/triplet_margin_loss.rb
 - lib/torch/nn/unfold.rb
+- lib/torch/nn/upsample.rb
 - lib/torch/nn/utils.rb
 - lib/torch/nn/weighted_loss.rb
 - lib/torch/nn/zero_pad2d.rb