Added PyTorch link and unit tests for argmax

[HangenYuu]

Description

Implemented the PyTorch link and unit tests for the Argmax Op.

Related Issue

• Closes #
• Related to Implement all Ops in PyTorch (help welcome!) #821

Checklist

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

Type of change

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

[ricardoV94]

Can you make the test tensor a tensor3 (3 dims), and test also axis=(0, 2)? This is something PyTensor supports but not e.g., numpy and perhaps pytorch as well. In that case you'll need to copy the logic used in perform instead of just calling argmax. If it works, great!

[ricardoV94]

In general with axis Ops we should always test axis=None, axis=(i,) and axis=(i, j), for 3d inputs, so we test all, a single axis or partial axes

[HangenYuu]

I will merge the content of this PR into #878.

[ricardoV94]

I will merge the content of this PR into #878.

As I wrote in the other PR this is not something we usually prefer. What was the reason that led you to do it?