# Last convolutionnal layer return 4D output
output = self.pool8(output)

# Rephase to a 2D element to work with Linear layers
output = output.view(output.size(0), 7*7*192)

# Do Linear operations now
output = self.fc9(output)