search

wavefrontshaping / complexPyTorch

A high-level toolbox for using complex valued neural networks in PyTorch
MIT License
610 stars 148 forks source link

A bug in ComplexBatchNorm1d #16

Closed yuzhiyang123 closed 2 years ago

yuzhiyang123 commented 2 years ago

In line 301 of ComplexLayers.py, it should be "input = input - mean[None, :]" instead of "input = input - mean[None, :, None, None]" since the input of ComplexBatchNorm1d only has 2 or 3 dimensions.

wavefrontshaping commented 2 years ago

Corrected, I replaced by input = input - mean[None, ...]

Thanks for notifying the error