Add flatten kwarg to FrameStackingWrapper to merge channel and stack dims.
Add flatten kwarg to FrameStackingWrapper to merge channel and stack dims.

PiperOrigin-RevId: 424580895
Change-Id: Ida38f025dc9318cdccfc0caf55be3b2450870b8d

deepmind/acme,deepmind/acme