使用gym.spacesMultiDiscrete()来生成多元离散空间
发布时间：2023-12-19 03:14:03
gym.spaces.MultiDiscrete是OpenAI Gym中的一个用于定义多元离散空间的类。它允许我们指定每个维度的离散取值范围。下面是一个使用gym.spaces.MultiDiscrete的示例代码：
import gym # 创建一个三维离散空间，每个维度的取值范围分别为[0, 5]、[0, 3]和[0, 2] action_space = gym.spaces.MultiDiscrete([6, 4, 3]) # 打印出维度和取值范围 print(action_space.nvec) # 随机产生一个动作 action = action_space.sample() # 打印出产生的动作 print(action)
以上代码创建了一个三维的离散空间，每个维度的取值范围分别为[0, 5]、[0, 3]和[0, 2]。在代码中，通过gym.spaces.MultiDiscrete([6, 4, 3])来创建这个离散空间，其中[6, 4, 3]表示了每个维度的取值范围。
可以通过打印action_space.nvec获取每个维度的取值范围。在上述代码中，它的输出为[6 4 3]。
可以使用action_space.sample()方法来随机产生一个动作。在上述代码中，他产生的动作可能是例如 [2 1 0]。
总之，gym.spaces.MultiDiscrete是一个非常有用的类，能够帮助我们定义多元离散空间，并方便地获取每个维度的取值范围以及随机生成动作。