Add docs for argmax_keepdim and specify what happens in the event of a tie

[michaeleisel]

trafficstars

argmax_keepdim has no documentation, which would be good to fix in general, but especially to resolve the question of what occurs when there are multiple elements along the target dimension that are tied for the max.