Sha256: dbd58520c6396da20769f934c275ce02b27b18f61e8aecd3b45e67a04d704d17
Contents?: true
Size: 288 Bytes
Versions: 86
Compression:
Stored size: 288 Bytes
Contents
module OnnxRuntime
module Utils
class << self
attr_accessor :mutex
end
self.mutex = Mutex.new
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
end
end
Version data entries
86 entries across 86 versions & 1 rubygems