Hey ahmad,
there is no need to specify an input layer, you can directly start using convolutional or linear layers in pytorch via torch.nn.Conv or torch.nn.Linear.
If you have any more questions, just post them.