nisharaichur:

self.conj = torch.tensor([1,-1,-1,-1], requires_grad=False)


Instead of: self.conj = torch.tensor([1,-1,-1,-1], requires_grad=False) perhaps:
self.register_buffer('conj ', torch.tensor([1,-1,-1,-1]))

So that it is transferred to gpu also?