Hi,
Is there any way that we don’t have to declare in-features of an nn.linear module explicitly? I am using Conv2d modules and I know that we need to use .view to flatten the values but can it be automatically done by extracting the incoming tensor shapes?