Class: Chainer::Initializers::Uniform

Inherits:
Chainer::Initializer show all
Defined in:
lib/chainer/initializers/uniform.rb

Instance Attribute Summary

Attributes inherited from Chainer::Initializer

#dtype

Instance Method Summary collapse

Constructor Details

#initialize(scale: 0.05, dtype: nil) ⇒ Uniform

Returns a new instance of Uniform.



4
5
6
7
 
# File 'lib/chainer/initializers/uniform.rb', line 4

def initialize(scale: 0.05, dtype: nil)
  @scale = scale
  super(dtype: dtype)
end

Instance Method Details

#call(array) ⇒ Object

Raises:

  • (ArgumentError) 


9
10
11
12
 
# File 'lib/chainer/initializers/uniform.rb', line 9

def call(array)
  raise ArgumentError.new("dtypes are missmatched. #{dtype} != #{array.class}") if dtype && dtype != array.class
  array.class.new(array.shape).rand(-@scale, @scale)
end