code: def space_shape(space):
      
      if isinstance(space, odl.ProductSpace) and space.is_power_space:
          return (len(space),) + space[0].shape
      else:
          return space.shape
docs: Return ``space.shape``, including power space base shape.
  
      If ``space`` is a power space, return ``(len(space),) + space[0].shape``,
      otherwise return ``space.shape``.