Match the output dimension of critic networks. For example, [128, 1] in PPO and [128] in Dreamer). We can reduce the last dimension.