Take the legal actions mask into account directly in the network.
Take the legal actions mask into account directly in the network.

This should speed up inference as it's not a separate step. It should also speed up training since it doesn't force the network to learn a valid policy for illegal actions that will be masked out anyway.

PiperOrigin-RevId: 298627991
Change-Id: Ib69ded2a1312b39152257ecf90ea74aa06bd220f

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