def __init__(self): self.gradInput = torch.Tensor() self.output = 0 self._backend = torch._thnn.type2backend[type(self.gradInput)]