增强: 支持网络类的批量输入

当我们有这样的环境空间时： self.observation_space = gym.spaces.Dict({ 'A': gym.spaces.Box(low=0, high=1, shape=(1,), dtype=np.float32), 'B': gym.spaces.Box(low=0, high=1, shape=(1,), dtype=np.float32), 'C': gym.spaces.Box(low=0, high=1, shape=(1,), dtype=np.float32), }) 我发现 tianshou.utils.net.common.Net 无法工作，因为它只支持以下两种类型作为输入： obs: np.ndarray | torch.Tensor，我想我们是否还可以添加对 Batch 的支持？