Generalize check_1d_shape to check_shape #2262

Job	Run time
pre-commit	45s
pin	2m 12s
coverage	1m 24s
benchmarks	1m 28s
test (ubuntu-latest, 3.11)	1m 11s
test (macos-latest, 3.11)	5m 23s
tutorials_list	5s
test (windows-latest, 3.11)	3m 35s
docs	1m 46s
examples	2m 44s
tutorials (tutorials/arm_repertoire.ipynb)	1m 6s
tutorials (tutorials/cma_mae.ipynb)	2m 9s
tutorials (tutorials/features/example.ipynb)	37s
tutorials (tutorials/fooling_mnist.ipynb)	2m 17s
tutorials (tutorials/lsi_mnist.ipynb)	2m 50s
tutorials (tutorials/lunar_lander.ipynb)	1m 43s
tutorials (tutorials/tom_cruise_dqd.ipynb)	6m 19s
deploy	0s
	37m 34s

Provide feedback