search

ankane / torch.rb

Deep learning for Ruby, powered by LibTorch
Other
704 stars 30 forks source link

Support for Complex numbers #25

Closed lfuszara1 closed 3 years ago

lfuszara1 commented 3 years ago 
irb(main):001:0> require 'torch.rb'
=> true
irb(main):002:0> Torch.tensor(1+1i)
/home/lukasz/.rbenv/versions/3.0.1/lib/ruby/gems/3.0.0/gems/torch-rb-0.5.3/lib/torch.rb:401:in `to_f': can't convert 1+1i into Float (RangeError)
    from /home/lukasz/.rbenv/versions/3.0.1/lib/ruby/gems/3.0.0/gems/torch-rb-0.5.3/lib/torch.rb:401:in `_tensor'
    from /home/lukasz/.rbenv/versions/3.0.1/lib/ruby/gems/3.0.0/gems/torch-rb-0.5.3/lib/torch.rb:401:in `tensor'

libtorch supports complex numbers but torch.rb not.

ankane commented 3 years ago

Hey @lfuszara1, added support on master.