Actually, the mean is calculated after each T iterations
for i in xrange(T):
    output_list.append(torch.unsqueeze(model(data), 0))
output_mean = torch.cat(output_list, 0).mean(0)

but by doing this for each image apart it really takes to much to run, so i was wondering if i could calculate the mean by batches ,will that make it run faster?