torch-rb 0.1.8 → 0.2.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (36) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +11 -2
  3. data/README.md +35 -11
  4. data/ext/torch/ext.cpp +37 -28
  5. data/ext/torch/extconf.rb +33 -6
  6. data/ext/torch/nn_functions.cpp +560 -0
  7. data/ext/torch/nn_functions.hpp +6 -0
  8. data/ext/torch/templates.hpp +2 -0
  9. data/ext/torch/tensor_functions.cpp +2085 -0
  10. data/ext/torch/tensor_functions.hpp +6 -0
  11. data/ext/torch/torch_functions.cpp +3175 -0
  12. data/ext/torch/torch_functions.hpp +6 -0
  13. data/lib/torch/ext.bundle +0 -0
  14. data/lib/torch/hub.rb +9 -0
  15. data/lib/torch/native/generator.rb +6 -3
  16. data/lib/torch/native/native_functions.yaml +539 -397
  17. data/lib/torch/native/parser.rb +2 -0
  18. data/lib/torch/nn/adaptive_avg_pool1d.rb +9 -0
  19. data/lib/torch/nn/adaptive_avg_pool2d.rb +9 -0
  20. data/lib/torch/nn/adaptive_avg_pool3d.rb +9 -0
  21. data/lib/torch/nn/adaptive_avg_poolnd.rb +14 -0
  22. data/lib/torch/nn/adaptive_max_pool1d.rb +9 -0
  23. data/lib/torch/nn/adaptive_max_pool2d.rb +9 -0
  24. data/lib/torch/nn/adaptive_max_pool3d.rb +9 -0
  25. data/lib/torch/nn/adaptive_max_poolnd.rb +15 -0
  26. data/lib/torch/nn/functional.rb +40 -2
  27. data/lib/torch/nn/module.rb +22 -1
  28. data/lib/torch/optim/lr_scheduler/cosine_annealing_lr.rb +29 -0
  29. data/lib/torch/optim/lr_scheduler/exponential_lr.rb +22 -0
  30. data/lib/torch/optim/lr_scheduler/lambda_lr.rb +28 -0
  31. data/lib/torch/optim/lr_scheduler/multi_step_lr.rb +23 -0
  32. data/lib/torch/optim/lr_scheduler/multiplicative_lr.rb +32 -0
  33. data/lib/torch/tensor.rb +8 -0
  34. data/lib/torch/version.rb +1 -1
  35. data/lib/torch.rb +21 -0
  36. metadata +38 -3
data/ext/torch/tensor_functions.hpp ADDED
@@ -0,0 +1,6 @@
1
+ // generated by rake generate:functions
2
+ // do not edit by hand
3
+
4
+ #pragma once
5
+
6
+ void add_tensor_functions(Module m);