RNaD algorithm: add chex asserts on shapes of the input arguments.
RNaD algorithm: add chex asserts on shapes of the input arguments.

PiperOrigin-RevId: 478984650
Change-Id: I55e225c4dc92e2f3e8ebb43e836047df67685270

deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel