torch-rb 0.19.1 → 0.21.0

data/lib/torch/nn/rnn_base.rb

@@ -1,9 +1,16 @@
 module Torch
   module NN
     class RNNBase < Module
-      def initialize(mode, input_size, hidden_size, num_layers: 1, bias: true,
-        batch_first: false, dropout: 0.0, bidirectional: false)
-
+      def initialize(
+        mode,
+        input_size,
+        hidden_size,
+        num_layers: 1,
+        bias: true,
+        batch_first: false,
+        dropout: 0.0,
+        bidirectional: false
+      )
         super()
         @mode = mode
         @input_size = input_size

data/lib/torch/nn/transformer.rb

@@ -7,13 +7,18 @@ module Torch
   module NN
     class Transformer < Module
       def initialize(
-        d_model: 512, nhead: 8,
-        num_encoder_layers: 6, num_decoder_layers: 6,
-        dim_feedforward: 2048, dropout: 0.1, activation: :relu,
-        custom_encoder: nil, custom_decoder: nil,
-        layer_norm_eps: 1e-5, batch_first: false
+        d_model: 512,
+        nhead: 8,
+        num_encoder_layers: 6,
+        num_decoder_layers: 6,
+        dim_feedforward: 2048,
+        dropout: 0.1,
+        activation: :relu,
+        custom_encoder: nil,
+        custom_decoder: nil,
+        layer_norm_eps: 1e-5,
+        batch_first: false
       )
-
         super()
 
         @encoder =
@@ -60,11 +65,15 @@ module Torch
       end
 
       def forward(
-        src, tgt,
-        src_mask: nil, tgt_mask: nil, memory_mask: nil,
-        src_key_padding_mask: nil, tgt_key_padding_mask: nil, memory_key_padding_mask: nil
+        src,
+        tgt,
+        src_mask: nil,
+        tgt_mask: nil,
+        memory_mask: nil,
+        src_key_padding_mask: nil,
+        tgt_key_padding_mask: nil,
+        memory_key_padding_mask: nil
       )
-
         if (!batch_first? && src.size(1) != tgt.size(1)) ||
           (batch_first? && src.size(0) != tgt.size(0))
 

data/lib/torch/nn/transformer_decoder_layer.rb

@@ -2,11 +2,14 @@ module Torch
   module NN
     class TransformerDecoderLayer < Module
       def initialize(
-        d_model, n_head,
-        dim_feedforward: 2048, dropout: 0.1, activation: :relu,
-        layer_norm_eps: 1e-5, batch_first: false
+        d_model,
+        n_head,
+        dim_feedforward: 2048,
+        dropout: 0.1,
+        activation: :relu,
+        layer_norm_eps: 1e-5,
+        batch_first: false
       )
-
         super()
 
         @self_attn = MultiheadAttention.new(d_model, n_head, dropout: dropout, batch_first: batch_first)

data/lib/torch/nn/transformer_encoder_layer.rb

@@ -2,11 +2,14 @@ module Torch
   module NN
     class TransformerEncoderLayer < Module
       def initialize(
-        d_model, n_head,
-        dim_feedforward: 2048, dropout: 0.1, activation: :relu,
-        layer_norm_eps: 1e-5, batch_first: false
+        d_model,
+        n_head,
+        dim_feedforward: 2048,
+        dropout: 0.1,
+        activation: :relu,
+        layer_norm_eps: 1e-5,
+        batch_first: false
       )
-
         super()
 
         @self_attn = MultiheadAttention.new(d_model, n_head, dropout: dropout, batch_first: batch_first)

data/lib/torch/tensor.rb

@@ -159,6 +159,7 @@ module Torch
 
     # TODO better compare?
     def <=>(other)
+      other = other.item if other.is_a?(Tensor)
       item <=> other
     end
 

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.19.1"
+  VERSION = "0.21.0"
 end

data/lib/torch.rb

@@ -13,6 +13,12 @@ require_relative "torch/inspector"
 require_relative "torch/tensor"
 require_relative "torch/version"
 
+# distributions
+require_relative "torch/distributions/distribution"
+require_relative "torch/distributions/exponential_family"
+require_relative "torch/distributions/normal"
+require_relative "torch/distributions/utils"
+
 # optim
 require_relative "torch/optim/optimizer"
 require_relative "torch/optim/adadelta"
@@ -433,7 +439,7 @@ module Torch
       # TODO check each dimensions for consistency in future
       raise Error, "Inconsistent dimensions" if data.size != size.inject(1, :*)
 
-      # TOOD move to C++
+      # TODO move to C++
       data = data.map { |v| v ? 1 : 0 } if options[:dtype] == :bool
 
       _tensor(data, size, tensor_options(**options))

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.19.1
+  version: 0.21.0
 platform: ruby
 authors:
 - Andrew Kane
 bindir: bin
 cert_chain: []
-date: 2025-02-10 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -64,6 +64,10 @@ files:
 - lib/torch-rb.rb
 - lib/torch.rb
 - lib/torch/device.rb
+- lib/torch/distributions/distribution.rb
+- lib/torch/distributions/exponential_family.rb
+- lib/torch/distributions/normal.rb
+- lib/torch/distributions/utils.rb
 - lib/torch/hub.rb
 - lib/torch/inspector.rb
 - lib/torch/nn/adaptive_avg_pool1d.rb
@@ -230,14 +234,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '3.1'
+      version: '3.2'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.2
+rubygems_version: 3.6.9
 specification_version: 4
 summary: Deep learning for Ruby, powered by LibTorch
 test_files: []