Module: Daimond

Defined in:
lib/daimond.rb,
lib/daimond/tensor.rb,
lib/daimond/version.rb,
lib/daimond/nn/conv2d.rb,
lib/daimond/nn/linear.rb,
lib/daimond/nn/module.rb,
lib/daimond/optim/sgd.rb,
lib/daimond/data/mnist.rb,
lib/daimond/nn/flatten.rb,
lib/daimond/optim/adam.rb,
lib/daimond/rust_bridge.rb,
lib/daimond/rust_backend.rb,
lib/daimond/nn/max_pool2d.rb,
lib/daimond/nn/conv2d_rust.rb,
lib/daimond/data/data_loader.rb,
lib/daimond/loss/cross_entropy.rb,
lib/daimond/nn/max_pool2d_rust.rb,
lib/daimond/utils/training_logger.rb

Defined Under Namespace

Modules: Data, Loss, NN, Optim, RustBackend, Utils Classes: Tensor

Constant Summary collapse

VERSION = 
"0.1.0"

Class Method Summary collapse

Class Method Details

.randn(*args) ⇒ Object 



27
28
29
 
# File 'lib/daimond.rb', line 27

def self.randn(*args)
  Tensor.randn(*args)
end

.rust_available? ⇒ Boolean

Проверяем загрузилась ли Rust библиотека

Returns:

  • (Boolean) 


5
6
7
 
# File 'lib/daimond/rust_bridge.rb', line 5

def self.rust_available?
  defined?(Daimond::Rust) && Daimond::Rust.respond_to?(:fast_matmul_flat)
end

.zeros(*args) ⇒ Object 



31
32
33
 
# File 'lib/daimond.rb', line 31

def self.zeros(*args)
  Tensor.zeros(*args)
end