Answer Prediction Code

Hi, I want to know why you stacked the 0th dimension when predicting the answer ？
`# final answer prediction `
`fusion_proj = torch.sum(torch.stack([out[:, 0, :] for out in out_list], dim=-1), dim=-1) `
`fusion_proj = self.vqproj(fusion_proj)`
