Add a note on axis arg of tf.argmax
I don't think we should clutter our documentation with statements like this.  Using a bool won't cause a crash, and shouldn't  cause confusion to a legitimate use-case.  This is just a case of someone bombarding the API with different inputs to see what happens.
As per documentation of he API tf.argmax the argument axis accepts integer values.If we pass any data types it is also accepting the same.Hence adding a note on same in the documentation to avoid any confusion as reported by the user in 
#63031.

shall fix #63031