Hi,
I know PyTorch layers expects an additional dimension for batch.
I was wondering how to extend this to a generic nn.Module (with no assumption about what are there inside), since applying a for loop over the batch duplicates the network (generating a siamese one).
Thanks in advance