Module: Chainer::Utils::Array

Defined in:
lib/chainer/utils/array.rb

Class Method Summary collapse

Class Method Details

.force_array(x, dtype = nil) ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
# File 'lib/chainer/utils/array.rb', line 4

def self.force_array(x, dtype=nil)
  if x.is_a? Integer or x.is_a? Float
    if dtype.nil?
      Numo::NArray.cast(x)
    else
      dtype.cast(x.dup)
    end
  else
    if dtype.nil?
      x
    else
      dtype.cast(x)
    end
  end
end