Method: Informers::Utils.ones_like

Defined in:
lib/informers/utils/tensor.rb

.ones_like(tensor) ⇒ Object 



30
31
32
33
34
35
 
# File 'lib/informers/utils/tensor.rb', line 30

def self.ones_like(tensor)
  if tensor[0].is_a?(Array)
    return tensor.map { |v| ones_like(v) }
  end
  tensor.map { |_| 1 }
end