torch-rb 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9667f9d3256f5e2d39937f17ae8eb00449dd14f79bb01cd647800bd7ed972fc6
-  data.tar.gz: 54c23612c79355e09c97da5fcf6b97c183da8316d1c2a53d6f8f0463e98342a2
+  metadata.gz: 51bcc56112e13ba206402857b379aee0df4c7695f75af354e833760adec67756
+  data.tar.gz: b2ff24940e4c219d88c5a001d4e8b4e44d0e55a35fc266989f0196e696d15bc8
 SHA512:
-  metadata.gz: bb2c8e5aae436367aeb871a2d19958e59ed9e9c7601b1b8b4473e33094cadf6d657947582b0ec93a29cb08723f8f7c81178a2d50beb23a125d5a356769d92177
-  data.tar.gz: 62feef39da31a19415e2e6c453aed4972e34db7367161a088944c06a977637a8b25cecc8eb2ad052b3b9deee0707f364e616cc33e7674cf0314899421f18fbee
+  metadata.gz: 95506016db5598333f0cb99a435d29951342af91f75ae4b1f01ef11df81891738888b90c7d27317071ad00bd9b81714cf41c0ea635c2578fd756c388b5e1da7f
+  data.tar.gz: 053c9c75e66fe54902f07413687deb6996afc7ae88217bd5dcc852ca59d535c663bb9fb3aed28b20dba953a42e714410867dbd6ecd747f96fe8e8dfd81da8d6c

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.1.7 (2019-01-10)
+
+- Fixed installation error with Ruby 2.7
+
 ## 0.1.6 (2019-12-09)
 
 - Added recurrent layers

data/ext/torch/ext.cpp

@@ -105,6 +105,13 @@ void Init_ext()
         return torch::zeros(size, options);
       })
     // begin operations
+    .define_singleton_method(
+      "_save",
+      *[](const Tensor &value) {
+        auto v = torch::pickle_save(value);
+        std::string str(v.begin(), v.end());
+        return str;
+      })
     .define_singleton_method(
       "_binary_cross_entropy_with_logits",
       *[](const Tensor &input, const Tensor &target, OptionalTensor weight, OptionalTensor pos_weight, MyReduction reduction) {

data/ext/torch/templates.hpp

@@ -5,62 +5,6 @@
 
 using namespace Rice;
 
-template<>
-inline
-long long from_ruby<long long>(Object x)
-{
-  return NUM2LL(x);
-}
-
-template<>
-inline
-Object to_ruby<long long>(long long const & x)
-{
-  return LL2NUM(x);
-}
-
-template<>
-inline
-unsigned long long from_ruby<unsigned long long>(Object x)
-{
-  return NUM2ULL(x);
-}
-
-template<>
-inline
-Object to_ruby<unsigned long long>(unsigned long long const & x)
-{
-  return ULL2NUM(x);
-}
-
-template<>
-inline
-short from_ruby<short>(Object x)
-{
-  return NUM2SHORT(x);
-}
-
-template<>
-inline
-Object to_ruby<short>(short const & x)
-{
-  return INT2NUM(x);
-}
-
-template<>
-inline
-unsigned short from_ruby<unsigned short>(Object x)
-{
-  return NUM2USHORT(x);
-}
-
-template<>
-inline
-Object to_ruby<unsigned short>(unsigned short const & x)
-{
-  return UINT2NUM(x);
-}
-
 // need to wrap torch::IntArrayRef() since
 // it doesn't own underlying data
 class IntArrayRef {

data/lib/torch.rb

@@ -300,6 +300,15 @@ module Torch
       Device.new(str)
     end
 
+    def save(obj, f)
+      raise NotImplementedYet unless obj.is_a?(Tensor)
+      File.binwrite(f, _save(obj))
+    end
+
+    def load(f)
+      raise NotImplementedYet
+    end
+
     # --- begin tensor creation: https://pytorch.org/cppdocs/notes/tensor_creation.html ---
 
     def arange(start, finish = nil, step = 1, **options)
@@ -383,19 +392,19 @@ module Torch
     # --- begin like ---
 
     def ones_like(input, **options)
-      ones(input.size, like_options(input, options))
+      ones(input.size, **like_options(input, options))
     end
 
     def empty_like(input, **options)
-      empty(input.size, like_options(input, options))
+      empty(input.size, **like_options(input, options))
     end
 
     def full_like(input, fill_value, **options)
-      full(input.size, fill_value, like_options(input, options))
+      full(input.size, fill_value, **like_options(input, options))
     end
 
     def rand_like(input, **options)
-      rand(input.size, like_options(input, options))
+      rand(input.size, **like_options(input, options))
     end
 
     def randint_like(input, low, high = nil, **options)
@@ -404,15 +413,15 @@ module Torch
         high = low
         low = 0
       end
-      randint(low, high, input.size, like_options(input, options))
+      randint(low, high, input.size, **like_options(input, options))
     end
 
     def randn_like(input, **options)
-      randn(input.size, like_options(input, options))
+      randn(input.size, **like_options(input, options))
     end
 
     def zeros_like(input, **options)
-      zeros(input.size, like_options(input, options))
+      zeros(input.size, **like_options(input, options))
     end
 
     private

data/lib/torch/nn/module.rb

@@ -79,11 +79,19 @@ module Torch
         _apply(convert)
       end
 
-      def call(*input)
-        forward(*input)
+      def call(*input, **kwargs)
+        forward(*input, **kwargs)
       end
 
-      def state_dict
+      def state_dict(destination: nil)
+        destination ||= {}
+        named_parameters.each do |k, v|
+          destination[k] = v
+        end
+        destination
+      end
+
+      def load_state_dict(state_dict)
         raise NotImplementedYet
       end
 

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.1.6"
+  VERSION = "0.1.7"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-12-10 00:00:00.000000000 Z
+date: 2020-01-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '2.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '2.2'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -261,7 +261,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubygems_version: 3.1.2
 signing_key: 
 specification_version: 4
 summary: Deep learning for Ruby, powered by LibTorch