Method: TensorFlow::Math.reduce_prod

Defined in:
lib/tensorflow/math.rb

.reduce_prod(input_tensor, axis: nil, keepdims: false) ⇒ Object 



338
339
340
341
342
 
# File 'lib/tensorflow/math.rb', line 338

def reduce_prod(input_tensor, axis: nil, keepdims: false)
  input_tensor = TensorFlow.convert_to_tensor(input_tensor)
  axis ||= reduction_dims(input_tensor)
  RawOps.prod(input: input_tensor, reduction_indices: axis, keep_dims: keepdims)
end