Revert "Update `amax/amin/norm/count_nonzero` signatures with `int[*]? dim` (#83300)"
This reverts commit 8c7c7ed3221aeeefb63ef2b7a221a5d8b274cda5.
Reverted https://github.com/pytorch/pytorch/pull/83300 on behalf of https://github.com/huydhn due to The commit pin breaks XLA test somehow