fix: preserve 2D outputs for (L,1) inputs#1137

MilesCranmerBot wants to merge 8 commits into

astroautomata:masterastroautomata/PySR:masterfrom

MilesCranmerBot:fix/torch-export-preserve-dimsMilesCranmerBot/PySR:fix/torch-export-preserve-dimsCopy head branch name to clipboard