Fix shapes in demo colab comments.
Fix shapes in demo colab comments.

PiperOrigin-RevId: 382377021
Change-Id: Ief4cc378c746200e2956bf0a0374b281ccbb82ec

deepmind/reverb,deepmind/reverb,deepmind/reverb,deepmind/reverb