Add options for scale_init and bias_init to SpinSphericalBatchNormalization.
Add options for scale_init and bias_init to SpinSphericalBatchNormalization.

Also refactor it to a cleaner implementation that avoids for loop and
concatenation.

PiperOrigin-RevId: 469988524

google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research