complexPyTorch
complexPyTorch copied to clipboard
Another batch norm suggestion
Hi, About the naive batch norm, I believe it's more effective to normalize the tensor with the absolute value rather than normalizing each part individually: https://github.com/wavefrontshaping/complexPyTorch/blob/2044cb077b3f139d59dff56abc378b1457de40d6/complexPyTorch/complexLayers.py#L213
What do you think?