fix for Zygote and ChainRules OneElement #3650

Job	Run time
test (1.10, ubuntu-latest, core_layers)	16m 52s
downgrade	1h 8m 25s
test (1.10, ubuntu-latest, normalize_layers)	13m 24s
test (1.10, ubuntu-latest, recurrent_layers)	19m 34s
test (1.10, ubuntu-latest, autodiff)	23m 28s
test (1.10, ubuntu-latest, misc)	32m 35s
test (1.10, ubuntu-latest, reactant)	10m 20s
test (1, ubuntu-latest, core_layers)	19m 44s
test (1, ubuntu-latest, normalize_layers)	17m 22s
test (1, ubuntu-latest, recurrent_layers)	23m 25s
test (1, ubuntu-latest, autodiff)	25m 3s
test (1, ubuntu-latest, misc)	34m 55s
test (1, ubuntu-latest, reactant)	15m 42s
test (1.10, macos-latest, all)	4h 7m 19s
test (1.10, windows-latest, all)	1h 15m 14s
	10h 43m 22s

Provide feedback