Fix bug in random policy when the arm features are structured.
Fix bug in random policy when the arm features are structured.

PiperOrigin-RevId: 311777774
Change-Id: I8b27136c3d12f58e77fc3401f1feb3b582b05a96

tensorflow/agents,tensorflow/agents