fix for Zygote and ChainRules OneElement #3643

Job	Run time
downgrade	1h 6m 8s
test (1.10, ubuntu-latest, core_layers)	15m 39s
test (1.10, ubuntu-latest, normalize_layers)	13m 21s
test (1.10, ubuntu-latest, recurrent_layers)	18m 33s
test (1.10, ubuntu-latest, autodiff)	23m 11s
test (1.10, ubuntu-latest, misc)	29m 20s
test (1.10, ubuntu-latest, reactant)	9m 48s
test (1, ubuntu-latest, core_layers)	19m 34s
test (1, ubuntu-latest, normalize_layers)	16m 30s
test (1, ubuntu-latest, recurrent_layers)	22m 35s
test (1, ubuntu-latest, autodiff)	24m 30s
test (1, ubuntu-latest, misc)	34m 27s
test (1, ubuntu-latest, reactant)	14m 54s
test (1.10, macos-latest, all)	3h 44m 39s
test (1.10, windows-latest, all)	1h 16m 8s
	10h 9m 17s

Provide feedback