torch-rb 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b37dc4dd7be5806879c2fb5bb52ac94c8b16eba76bc2a3c591ca4cbe51cf8745
-  data.tar.gz: 6415d14f7cc8baa4db9205c709b70fec2b90b4dff1b60de97c299c2a7edfbf40
+  metadata.gz: 8e1f9758c937519ca31d92f3acd35ce0372f8cf57362cd3b50bd45920e7a6763
+  data.tar.gz: 4d370857ee758694b974da0f5d0973a687181ef2fedcad5c583f446ceb67dda2
 SHA512:
-  metadata.gz: cf43cb21e18171f76f1291f2cccdb8a93141605fe33d4421eaf799a4589638d33da040b2ffad3ddce34fec60ab2b41edf1fa9a247d69de7f31e157063e57f331
-  data.tar.gz: 3c858de8e7eb6169359fad18104c08d9a4011a305f44c4a8213a289b005dc4439b1959a40effd8b1bba1c2e96a871b28f01695fdda56e6ad0aeed0c2334cfa25
+  metadata.gz: b1cbb37019852bfdbfc45b28ac32924b4de313ce21112ffe8bb5ec91fe17898d3a1ceb42d77540e6b7a0d656e9443002fb72a1201453507dca4915db13879167
+  data.tar.gz: d1a35689c3ad6a0628485633af4f6d7f613288fbf739f9e94ccfdb72c613b0d4581a21e00aef3a309771967f65086034befc519bc89ecc7884ff1dd142a8289f

data/CHANGELOG.md

@@ -1,3 +1,8 @@
+## 0.2.2 (2020-04-27)
+
+- Added support for saving tensor lists
+- Added `ndim` and `ndimension` methods to tensors
+
 ## 0.2.1 (2020-04-26)
 
 - Added support for saving and loading models

data/README.md

@@ -426,7 +426,7 @@ Torch::CUDA.available?
 Move a neural network to a GPU
 
 ```ruby
-net.to("cuda")
+net.cuda
 ```
 
 ## rbenv

data/ext/torch/ext.cpp

@@ -82,6 +82,16 @@ void Init_ext()
       *[](torch::IValue& self) {
         return self.toInt();
       })
+    .define_method(
+      "to_list",
+      *[](torch::IValue& self) {
+        auto list = self.toListRef();
+        Array obj;
+        for (auto& elem : list) {
+          obj.push(to_ruby<torch::IValue>(torch::IValue{elem}));
+        }
+        return obj;
+      })
     .define_method(
       "to_string_ref",
       *[](torch::IValue& self) {
@@ -96,17 +106,27 @@ void Init_ext()
       "to_generic_dict",
       *[](torch::IValue& self) {
         auto dict = self.toGenericDict();
-        Hash h;
+        Hash obj;
         for (auto& pair : dict) {
-          h[to_ruby<torch::IValue>(torch::IValue{pair.key()})] = to_ruby<torch::IValue>(torch::IValue{pair.value()});
+          obj[to_ruby<torch::IValue>(torch::IValue{pair.key()})] = to_ruby<torch::IValue>(torch::IValue{pair.value()});
         }
-        return h;
+        return obj;
       })
     .define_singleton_method(
       "from_tensor",
       *[](torch::Tensor& v) {
         return torch::IValue(v);
       })
+    // TODO create specialized list types?
+    .define_singleton_method(
+      "from_list",
+      *[](Array obj) {
+        c10::impl::GenericList list(c10::AnyType::get());
+        for (auto entry : obj) {
+          list.push_back(from_ruby<torch::IValue>(entry));
+        }
+        return torch::IValue(list);
+      })
     .define_singleton_method(
       "from_string",
       *[](String v) {

data/lib/torch/tensor.rb

@@ -4,6 +4,8 @@ module Torch
     include Inspector
 
     alias_method :requires_grad?, :requires_grad
+    alias_method :ndim, :dim
+    alias_method :ndimension, :dim
 
     def self.new(*args)
       FloatTensor.new(*args)
@@ -143,6 +145,7 @@ module Torch
       neg
     end
 
+    # TODO better compare?
     def <=>(other)
       item <=> other
     end

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.2.1"
+  VERSION = "0.2.2"
 end

data/lib/torch.rb

@@ -466,6 +466,12 @@ module Torch
         IValue.from_bool(obj)
       when nil
         IValue.new
+      when Array
+        if obj.all? { |v| v.is_a?(Tensor) }
+          IValue.from_list(obj.map { |v| IValue.from_tensor(v) })
+        else
+          raise Error, "Unknown list type"
+        end
       else
         raise Error, "Unknown type: #{obj.class.name}"
       end
@@ -490,6 +496,8 @@ module Torch
           dict[to_ruby(k)] = to_ruby(v)
         end
         dict
+      elsif ivalue.list?
+        ivalue.to_list.map { |v| to_ruby(v) }
       else
         type =
           if ivalue.capsule?
@@ -510,8 +518,6 @@ module Torch
             "BoolList"
           elsif ivalue.tensor_list?
             "TensorList"
-          elsif ivalue.list?
-            "List"
           elsif ivalue.object?
             "Object"
           elsif ivalue.module?

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - Andrew Kane