Module: OnnxRuntime::Utils

Defined in:
lib/onnxruntime/utils.rb

Class Attribute Summary collapse

Class Method Summary collapse

Class Attribute Details

.mutex ⇒ Object

Returns the value of attribute mutex.



4
5
6
 
# File 'lib/onnxruntime/utils.rb', line 4

def mutex
  @mutex
end

Class Method Details

.reshape(arr, dims) ⇒ Object 



8
9
10
11
12
13
14
 
# File 'lib/onnxruntime/utils.rb', line 8

def self.reshape(arr, dims)
  arr = arr.flatten
  dims[1..-1].reverse.each do |dim|
    arr = arr.each_slice(dim)
  end
  arr.to_a
end