Part 1: Lesson3 predictions from learn.TTA() return weird shape

[laphi]

See this thread

learn.TTA() returns a 3-d dataset for log_preds

When I run -> log_preds, y = learn.TTA(), I get a 3-d array for log_preds. I see from lecture videos that it should be a 2-d array. What am I missing? Thank you in advance.

and use the solution by eduardopoleo (take the average).