No, we require providing the exact numbers of features at construction time. I think it’s safer and you gain more insight into your network structure, because you know exactly what kind of dimensionalities are used. If it’s really unacceptable for you, you might create the layers lazily in the forward function, but that’s not recommended.