I’m sorry for asking so many questions, but I’m a bit confused about this method. I would like to know why they are calculating the mean in:
output_mean = torch.cat(output_list, 0).mean(0)

for just the first element of the output_list  and considering it as the correct prediction from  all the T tensors found in the output_list