Method: TensorFlow::Math.reduce_max
- Defined in:
- lib/tensorflow/math.rb
.reduce_max(input_tensor, axis: nil, keepdims: false) ⇒ Object
def reduce_logsumexp end
320 321 322 323 324 |
# File 'lib/tensorflow/math.rb', line 320 def reduce_max(input_tensor, axis: nil, keepdims: false) input_tensor = TensorFlow.convert_to_tensor(input_tensor) axis ||= reduction_dims(input_tensor) RawOps.max(input: input_tensor, reduction_indices: axis, keep_dims: keepdims) end |