(Solution code removed, as posting it publicly is against the honour code of this community, regardless if it is correct or not)
Can anyone let me know what is the issue of my code?
Thank you so much!
You didn’t set the axis parameter for tl.Mean()
Thank you! I think I found the reason. I need to pass in the parameters in the EncoderBlocks when it is in different head.