Module: Differentiation
- Defined in:
- lib/differentiation/dual_number.rb,
lib/differentiation.rb,
lib/differentiation/version.rb
Overview
fronzen_string_literal: true
Defined Under Namespace
Classes: DualNumber
Constant Summary collapse
- VERSION =
"0.2.2"
Class Method Summary collapse
- .convert_to_dual_number(o, key: nil, named_variables: {}) ⇒ Object
- .differentiable?(o) ⇒ Boolean
- .differential(f) ⇒ Object
Class Method Details
.convert_to_dual_number(o, key: nil, named_variables: {}) ⇒ Object
11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
# File 'lib/differentiation.rb', line 11 def self.convert_to_dual_number(o, key: nil, named_variables: {}) if o.is_a?(DualNumber) if key and o.named_variables.empty? DualNumber.new(o.n, o.diff, key: key) else o end elsif differentiable?(o) if defined?(Matrix) and o.is_a?(Matrix) if key vars = Matrix.build(o.row_size, o.column_size){ nil } named_variables = { key => vars } end Matrix.build(o.row_size, o.column_size) do |i, j| v = self.convert_to_dual_number(o[i, j], named_variables: named_variables) if key vars.__send__(:[]=, i, j, v) end v end else DualNumber.new(o, key: key, named_variables: named_variables) end else o end end |
.differentiable?(o) ⇒ Boolean
7 8 9 |
# File 'lib/differentiation.rb', line 7 def self.differentiable?(o) o.is_a?(Numeric) or (defined?(Matrix) and o.is_a?(Matrix)) end |
.differential(f) ⇒ Object
39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
# File 'lib/differentiation.rb', line 39 def self.differential(f) unless f.is_a?(Proc) or f.is_a?(Method) or f.is_a?(UnboundMethod) raise TypeError, "Only Proc or Method can be differentiable" end # [[:req, :x], [:req, :y], [:opt, :c], [:rest, :rest], [:keyreq, :k2], [:key, :k1], [:keyrest, :kw]] parameters = f.parameters positional = [] parameters.each do |tuple| if [:req, :opt].include?(tuple[0]) positional << tuple[1] elsif [:key, :keyreq, :kwrest].include?(tuple[0]) else # TODO: support rest arguments raise "rest arguments are currently not supported" end end f_prime = lambda {|*args, **kwargs| if f.is_a?(UnboundMethod) f = f.bind(self) end args.map!.with_index do |a, i| Differentiation.convert_to_dual_number(a, key: positional[i]) end kwargs = Hash[kwargs.map{|k,v| [k, Differentiation.convert_to_dual_number(v, key: k)] }] if kwargs.empty? Differentiation.convert_to_dual_number(f.call(*args)) else Differentiation.convert_to_dual_number(f.call(*args, **kwargs)) end } end |