I’m not sure I’m understanding the question completely, but if you would like to print the current input and parameter device, you could add:
def forward(self, x):
    print(x.device, self.fc.weight.device, ...)

to your forward method.