Update softmax_policy.py
Update softmax_policy.py

Applied the changes recommended by sgirgin.

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