features is a list! []
I’m a little confused when you say my features have the shape [10, X, 2048, 2, 4], because based on the printed output I wrote in my question, they have the shape [X,2048,2,4] don’t they?