Added PyTorch link and unit tests for argmax#900

HangenYuu wants to merge 1 commit intopymc-devs:mainpymc-devs/pytensor:mainfrom

HangenYuu:torch_argmaxHangenYuu/pytensor:torch_argmaxCopy head branch name to clipboard