Module: RubyZero

Includes:
Core::DataTypes
Defined in:
lib/rubyzero/nn/nn.rb,
lib/rubyzero/version.rb,
lib/rubyzero/core/core.rb,
lib/rubyzero/data/data.rb,
lib/rubyzero/core/device.rb,
lib/rubyzero/core/dtypes.rb,
lib/rubyzero/utils/utils.rb,
lib/rubyzero/nn/functional.rb,
lib/rubyzero/nn/layers/layers.rb,
lib/rubyzero/core/tensor_initialize_methods.rb

Defined Under Namespace

Modules: Core, Data, NN, Utils

Constant Summary collapse

VERSION = 
"0.1.5"
F = 
RubyZero::NN::Functional
L = 
RubyZero::NN::LayersInitializeAliases
FloatTensor = 
RubyZero::Core::DataTypes::Float32
DoubleTensor = 
RubyZero::Core::DataTypes::Float64
IntTensor = 
RubyZero::Core::DataTypes::Int32
LongTensor = 
RubyZero::Core::DataTypes::Int64
@@ident_device = 
{}

Class Method Summary collapse

Methods included from Core::DataTypes

from_xmo_dtype

Class Method Details

.device(identifier) ⇒ Object 



23
24
25
26
27
28
29
30
31
 
# File 'lib/rubyzero/core/device.rb', line 23

def self.device(identifier)
    if @@ident_device[identifier]
        return @@ident_device[identifier]
    else
        d = Core::Device.new(identifier)
        @@ident_device[identifier] = d
        return d
    end
end