lib/tensor_stream/generated_stub/ops.rb
def argmax(input_a, axis = nil, name: nil, dimension: nil, output_type: :int32)
      check_allowed_types(input_a, TensorStream::Ops::NUMERIC_TYPES)
      check_allowed_types(axis, TensorStream::Ops::INTEGER_TYPES)
      _op(:argmax, input_a, axis, name: name, dimension: dimension, output_type: output_type)
    end
def argmax(input_a, axis = nil, name: nil, dimension: nil, output_type: :int32)
      check_allowed_types(input_a, TensorStream::Ops::NUMERIC_TYPES)
      check_allowed_types(axis, TensorStream::Ops::INTEGER_TYPES)
      _op(:argmax, input_a, axis, name: name, dimension: dimension, output_type: output_type)
    end
Returns the index with the largest value across axes of a tensor.


 @param input_a tensor X (of type NUMERIC_TYPES)
 @param axis Describes which axis of the input tensor to reduce across. For vectors, use axis = 0 (of type INTEGER_TYPES)

 Options:
 @option name Optional name
 @option dimension Same as axis
 @option output_type Output data type defaults to int32 default (:int32)
 @return Tensor