I’m passing a noise input in the shape [50, 300] and expect the model to output a tensor as it returns a wrong tensor of [50,1,50,300]. Currently, I get [50, 1, 28, 28] which is why I get an error. (Updated the stack trace!)