tf_agents.bandits.specs.utils.create_per_arm_observation_spec

View source on GitHub

Creates an observation spec with per-arm features and possibly action mask.