torch-rb 0.4.1 → 0.5.3

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (23) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +24 -0
  3. data/README.md +9 -1
  4. data/codegen/generate_functions.rb +13 -8
  5. data/codegen/native_functions.yaml +2363 -714
  6. data/ext/torch/ext.cpp +15 -16
  7. data/ext/torch/ruby_arg_parser.h +26 -2
  8. data/ext/torch/templates.h +1 -0
  9. data/ext/torch/wrap_outputs.h +7 -0
  10. data/lib/torch.rb +5 -7
  11. data/lib/torch/nn/module.rb +101 -21
  12. data/lib/torch/optim/adadelta.rb +3 -3
  13. data/lib/torch/optim/adagrad.rb +3 -3
  14. data/lib/torch/optim/adam.rb +2 -2
  15. data/lib/torch/optim/adamax.rb +2 -2
  16. data/lib/torch/optim/adamw.rb +2 -2
  17. data/lib/torch/optim/asgd.rb +1 -1
  18. data/lib/torch/optim/rmsprop.rb +7 -7
  19. data/lib/torch/optim/rprop.rb +1 -1
  20. data/lib/torch/optim/sgd.rb +2 -2
  21. data/lib/torch/tensor.rb +5 -0
  22. data/lib/torch/version.rb +1 -1
  23. metadata +3 -3
data/lib/torch/optim/rprop.rb CHANGED
@@ -52,7 +52,7 @@ module Torch
52
52
  grad[sign.eq(etaminus)] = 0
53
53
 
54
54
  # update parameters
55
- p.data.addcmul!(-1, grad.sign, step_size)
55
+ p.data.addcmul!(grad.sign, step_size, value: -1)
56
56
 
57
57
  state[:prev].copy!(grad)
58
58
  end
data/lib/torch/optim/sgd.rb CHANGED
@@ -36,14 +36,14 @@ module Torch
36
36
  end
37
37
  if momentum != 0
38
38
  param_state = @state[p]
39
- if !param_state.key(:momentum_buffer)
39
+ if !param_state.key?(:momentum_buffer)
40
40
  buf = param_state[:momentum_buffer] = Torch.clone(d_p).detach
41
41
  else
42
42
  buf = param_state[:momentum_buffer]
43
43
  buf.mul!(momentum).add!(d_p, alpha: 1 - dampening)
44
44
  end
45
45
  if nesterov
46
- d_p = d_p.add(momentum, buf)
46
+ d_p = d_p.add(buf, alpha: momentum)
47
47
  else
48
48
  d_p = buf
49
49
  end
data/lib/torch/tensor.rb CHANGED
@@ -174,5 +174,10 @@ module Torch
174
174
  return _random!(0, *args) if args.size == 1
175
175
  _random!(*args)
176
176
  end
177
+
178
+ # center option
179
+ def stft(*args)
180
+ Torch.stft(*args)
181
+ end
177
182
  end
178
183
  end
data/lib/torch/version.rb CHANGED
@@ -1,3 +1,3 @@
1
1
  module Torch
2
- VERSION = "0.4.1"
2
+ VERSION = "0.5.3"
3
3
  end
metadata CHANGED
@@ -1,14 +1,14 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: torch-rb
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.4.1
4
+ version: 0.5.3
5
5
  platform: ruby
6
6
  authors:
7
7
  - Andrew Kane
8
8
  autorequire:
9
9
  bindir: bin
10
10
  cert_chain: []
11
- date: 2020-10-13 00:00:00.000000000 Z
11
+ date: 2021-01-14 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: rice
@@ -289,7 +289,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
289
289
  - !ruby/object:Gem::Version
290
290
  version: '0'
291
291
  requirements: []
292
- rubygems_version: 3.0.3
292
+ rubygems_version: 3.2.3
293
293
  signing_key:
294
294
  specification_version: 4
295
295
  summary: Deep learning for Ruby, powered by LibTorch