Class: Micrograd::Layer

Inherits:

Object

• Object
• Micrograd::Layer

Defined in:

lib/micrograd/layer.rb

Instance Attribute Summary

• #neurons ⇒ Object readonly

  Returns the value of attribute neurons.

Instance Method Summary

• #call(xs) ⇒ Object
• #initialize(inputs, outputs) ⇒ Layer constructor

  A new instance of Layer.

• #parameters ⇒ Object

Constructor Details

#initialize(inputs, outputs) ⇒ Layer

Returns a new instance of Layer.

# File 'lib/micrograd/layer.rb', line 5

def initialize(inputs, outputs)
  @neurons = Array.new(outputs) { Neuron.new(inputs) }
end

Instance Attribute Details

#neurons ⇒ Object (readonly)

Returns the value of attribute neurons.

# File 'lib/micrograd/layer.rb', line 9

def neurons
  @neurons
end

Instance Method Details

#call(xs) ⇒ Object

# File 'lib/micrograd/layer.rb', line 11

def call(xs)
  @neurons.map { |n| n.call(xs) }
end

#parameters ⇒ Object

# File 'lib/micrograd/layer.rb', line 15

def parameters = neurons.flat_map(&:parameters)