No no not at all, I should have been able to work that out for myself 
Okay that seems to be okay except that I get the following error from the .zeros
AttributeError: module 'torch.nn.init' has no attribute 'zeros_'
Perhaps it’s an outdated atribute?