torch-rb 0.9.2 → 0.10.0

data/ext/torch/sparse_functions.h

@@ -0,0 +1,6 @@
+// generated by rake generate:functions
+// do not edit by hand
+
+#pragma once
+
+void add_sparse_functions(Rice::Module& m);

data/lib/torch/nn/parameter_list.rb

@@ -0,0 +1,48 @@
+module Torch
+  module NN
+    class ParameterList < Module
+      include Enumerable
+
+      def initialize(parameters)
+        super()
+        @initialized = true
+        unless parameters.nil?
+          concat(parameters)
+        end
+      end
+
+      def length
+        @parameters.length
+      end
+      alias_method :count, :length
+      alias_method :size, :length
+
+      def concat(parameters)
+        unless parameters.is_a?(Enumerable)
+          raise TypeError, "ParameterList#concat should be called with an enumerable, but got #{parameters.class.name}"
+        end
+        offset = length
+        parameters.each_with_index do |param, i|
+          register_parameter((offset + i).to_s, param)
+        end
+        self
+      end
+
+      def each(&block)
+        if block_given?
+          @parameters.values.each(&block)
+        else
+          to_enum(:each)
+        end
+      end
+
+      def [](idx)
+        if idx.is_a?(Range)
+          self.class.new(@parameters.values[idx])
+        else
+          @parameters[idx.to_s]
+        end
+      end
+    end
+  end
+end

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.9.2"
+  VERSION = "0.10.0"
 end

data/lib/torch.rb

@@ -40,6 +40,7 @@ require "torch/nn/utils"
 # nn containers
 require "torch/nn/module"
 require "torch/nn/module_list"
+require "torch/nn/parameter_list"
 require "torch/nn/sequential"
 
 # nn convolution layers

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.9.2
+  version: 0.10.0
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-02-03 00:00:00.000000000 Z
+date: 2022-03-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -52,6 +52,7 @@ files:
 - ext/torch/random.cpp
 - ext/torch/ruby_arg_parser.cpp
 - ext/torch/ruby_arg_parser.h
+- ext/torch/sparse_functions.h
 - ext/torch/special.cpp
 - ext/torch/special_functions.h
 - ext/torch/templates.h
@@ -149,6 +150,7 @@ files:
 - lib/torch/nn/nll_loss.rb
 - lib/torch/nn/pairwise_distance.rb
 - lib/torch/nn/parameter.rb
+- lib/torch/nn/parameter_list.rb
 - lib/torch/nn/poisson_nll_loss.rb
 - lib/torch/nn/prelu.rb
 - lib/torch/nn/reflection_pad1d.rb
@@ -227,7 +229,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.3
+rubygems_version: 3.3.7
 signing_key:
 specification_version: 4
 summary: Deep learning for Ruby, powered by LibTorch