torch-rb 0.2.7 → 0.3.0

data/lib/torch/nn/conv2d.rb

@@ -18,7 +18,6 @@ module Torch
         F.conv2d(input, @weight, @bias, @stride, @padding, @dilation, @groups)
       end
 
-      # TODO add more parameters
       def extra_inspect
         s = String.new("%{in_channels}, %{out_channels}, kernel_size: %{kernel_size}, stride: %{stride}")
         s += ", padding: %{padding}" if @padding != [0] * @padding.size

data/lib/torch/optim/optimizer.rb

@@ -32,9 +32,11 @@ module Torch
       end
 
       def state_dict
+        raise NotImplementedYet
+
         pack_group = lambda do |group|
-          packed = group.select { |k, _| k != :params }.to_h
-          packed[:params] = group[:params].map { |p| p.object_id }
+          packed = group.select { |k, _| k != :params }.map { |k, v| [k.to_s, v] }.to_h
+          packed["params"] = group[:params].map { |p| p.object_id }
           packed
         end
 
@@ -42,8 +44,8 @@ module Torch
         packed_state = @state.map { |k, v| [k.is_a?(Tensor) ? k.object_id : k, v] }.to_h
 
         {
-          state: packed_state,
-          param_groups: param_groups
+          "state" => packed_state,
+          "param_groups" => param_groups
         }
       end
 

data/lib/torch/tensor.rb

@@ -47,10 +47,15 @@ module Torch
       end
     end
 
-    # TODO support dtype
-    def to(device, non_blocking: false, copy: false)
+    def to(device = nil, dtype: nil, non_blocking: false, copy: false)
+      device ||= self.device
       device = Device.new(device) if device.is_a?(String)
-      _to(device, _dtype, non_blocking, copy)
+
+      dtype ||= self.dtype
+      enum = DTYPE_TO_ENUM[dtype]
+      raise Error, "Unknown type: #{dtype}" unless enum
+
+      _to(device, enum, non_blocking, copy)
     end
 
     def cpu
@@ -215,11 +220,11 @@ module Torch
       value = Torch.tensor(value, dtype: dtype) unless value.is_a?(Tensor)
 
       if index.is_a?(Numeric)
-        copy_to(_select_int(0, index), value)
+        index_put!([Torch.tensor(index)], value)
       elsif index.is_a?(Range)
         finish = index.end
         finish += 1 unless index.exclude_end?
-        copy_to(_slice_tensor(0, index.begin, finish, 1), value)
+        _slice_tensor(0, index.begin, finish, 1).copy!(value)
       elsif index.is_a?(Tensor)
         index_put!([index], value)
       else
@@ -254,11 +259,5 @@ module Torch
       _clamp_min_(min)
       _clamp_max_(max)
     end
-
-    private
-
-    def copy_to(dst, src)
-      dst.copy!(src)
-    end
   end
 end

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.2.7"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.2.7
+  version: 0.3.0
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-06-30 00:00:00.000000000 Z
+date: 2020-07-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice