torch-rb 0.5.0 → 0.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fa3f2b23a31aeaec5f651db7bf30b91d4b3da4cd539645cf75b9bafa0888b1cf
-  data.tar.gz: 886cdc9f34a7ec5ed758f52c039d5d9fc6f0c7a51ee3bf6ef25f4382e9192225
+  metadata.gz: 68b4f1b0599e97803bd0e07efdd7fff96e0e04369005dbd72db190df3cf4e1b4
+  data.tar.gz: 302f75c8b43b25ac06e49c7d8047c755a58d38d86e67c6b8df4f00a8579a2676
 SHA512:
-  metadata.gz: 1683c37ed188b8ea505b4d2040b4c6f42c75b682a0251ada88982bb4c7b1318c8aa3ebc5b755eb7a46397f5cffcaa46af82e17207f402fcbd7152f44f2e28cab
-  data.tar.gz: 3ac8ba3b4f854f964a4ffaaa6628ac236d94bca3462aff6775858160beebb869c7823b0aaa0378572d061bc33059b1b5f0de0d8608107161463cff796b753a2d
+  metadata.gz: d18a01b8d18f659fb2da9e4cf6e700de260c04f7400ce62e5850d79f03f5e58e5b700730c7bdddab0148e60be080b5b85bb48e53527653372527461da46863d7
+  data.tar.gz: 8aaa9e2ee75e2a1a64ae506a59eca35e150bf48568dc4c905fb30df6a61d56ab745383f31525fd3c9e6a259cf33adf154ba8d70476b75fd124b035dbf9b48f4f

data/CHANGELOG.md

@@ -1,3 +1,8 @@
+## 0.5.1 (2020-10-28)
+
+- Fixed error with tensor classes and no arguments
+- Fixed error with `stft` and `clamp` methods
+
 ## 0.5.0 (2020-10-28)
 
 - Updated LibTorch to 1.7.0

data/codegen/generate_functions.rb

@@ -112,7 +112,7 @@ def generate_attach_def(name, type, def_method)
       name
     end
 
-  ruby_name = "_#{ruby_name}" if ["size", "stride", "random!"].include?(ruby_name)
+  ruby_name = "_#{ruby_name}" if ["size", "stride", "random!", "stft"].include?(ruby_name)
 
   # cast for Ruby < 2.7 https://github.com/thisMagpie/fftw/issues/22#issuecomment-49508900
   cast = RUBY_VERSION.to_f > 2.7 ? "" : "(VALUE (*)(...)) "

data/lib/torch.rb

@@ -261,6 +261,8 @@ module Torch
         Torch._from_blob(bytes, [bytes.bytesize], TensorOptions.new.dtype(DTYPE_TO_ENUM[dtype]))
       elsif args.size == 1 && args.first.is_a?(Array)
         Torch.tensor(args.first, dtype: dtype, device: device)
+      elsif args.size == 0
+        Torch.empty(0, dtype: dtype, device: device)
       else
         Torch.empty(*args, dtype: dtype, device: device)
       end
@@ -434,7 +436,8 @@ module Torch
       zeros(input.size, **like_options(input, options))
     end
 
-    def stft(input, n_fft, hop_length: nil, win_length: nil, window: nil, center: true, pad_mode: "reflect", normalized: false, onesided: true)
+    # center option
+    def stft(input, n_fft, hop_length: nil, win_length: nil, window: nil, center: true, pad_mode: "reflect", normalized: false, onesided: true, return_complex: nil)
       if center
         signal_dim = input.dim
         extended_shape = [1] * (3 - signal_dim) + input.size
@@ -442,12 +445,7 @@ module Torch
         input = NN::F.pad(input.view(extended_shape), [pad, pad], mode: pad_mode)
         input = input.view(input.shape[-signal_dim..-1])
       end
-      _stft(input, n_fft, hop_length, win_length, window, normalized, onesided)
-    end
-
-    def clamp(tensor, min, max)
-      tensor = _clamp_min(tensor, min)
-      _clamp_max(tensor, max)
+      _stft(input, n_fft, hop_length, win_length, window, normalized, onesided, return_complex)
     end
 
     private

data/lib/torch/tensor.rb

@@ -174,5 +174,10 @@ module Torch
       return _random!(0, *args) if args.size == 1
       _random!(*args)
     end
+
+    # center option
+    def stft(*args)
+      Torch.stft(*args)
+    end
   end
 end

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.5.0"
+  VERSION = "0.5.1"
 end

metadata

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