torch.argmax ============ .. currentmodule:: torch .. autofunction:: argmax