Hi there,
is there any way one can figure out the output dimension of a model without passing a sample to it?
For example, I have two network net1 and net2. I want to use another network net3, which maps the concatenation of net1 and net2 as the feature to some label. By defining the net3, I have to specify the input dimension which equals net1.output_size + net2.output_size.
My idea would be net1.modules[-1].out_features + net2.modules[-1].out_features. Is there a more elegant solution？
Thanks!