Break MaxandArgmax Op to seperate TensorMax Op and Argmax Op

[Dhruvanshu-Joshi]

Description

MaxandArgmax Op calculates both maximum and argmax together. With this PR, we aim to have seperate ops for the two operations.

Related Issue

• [x] Closes #334
• [ ] Related to #

Checklist

• [x] Checked that the pre-commit linting/style checks pass
• [x] Included tests that prove the fix is effective or that the new feature works
• [x] Added necessary documentation (docstrings and/or example notebooks)
• [ ] If you are a pro: each commit corresponds to a relevant logical change

Type of change

• [x] New feature / enhancement
• [ ] Bug fix
• [ ] Documentation
• [ ] Maintenance
• [ ] Other (please specify):