def space_shape(space):
"""return the shape of tensor expected for a given space"""
if isinstance(space, spaces.Discrete):
return [space.n]
else:
return space.shape
微信
公众号
扫码关注公众号