Is there a good way to get the output shape of a nn.Module object without knowing the input shape? Everything I can come up with seems to need a few extra assumptions on the structure of the network.
Thanks!