Revert "Update `amax/amin/norm/count_nonzero` signatures with `int[*]? dim` (#83300)"
This reverts commit 1c0f0b33a0e013d6ec162cf488ff7643c4ffa33e.
Reverted https://github.com/pytorch/pytorch/pull/83300 on behalf of https://github.com/jeffdaily due to The commit breaks nvfuser tests