address equations 4 and 5 in the soundstream paper - "multi spectral recon...

address equations 4 and 5 in the soundstream paper - "multi spectral recon loss", raised in issue https://github.com/lucidrains/audiolm-pytorch/issues/76