expand_dims eager optimization

[HangenYuu]

Description

A small eager optimization to avoid useless dimshuffles in expand_dims, follwing squeeze logic.

Related Issue

• Closes Avoid dimshuffle if expand_dims has empty axis #707
• Related to #

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]

@HangenYuu you seem to have started this from a branch other than main (it includes a lot of unrelated commits, from your other PR)

[HangenYuu]

Oh. I created a new branch from the main branch but turn out it's of my fork. Let me close this and re-fork it properly.