Thanks for the info.
Yes, that’s exactly right - I do not know (or atleast I want to assume that I do not know) the dimension of x.shape[1]. By precompute - do you mean like manually calculate the dimension and then just enter the appropriate dimensions in the __init__ function? (i.e when I am defining the individual modules like Convolutional or FC or whatever)