def forward(self, h: torch.Tensor, x_d: torch.Tensor):
    gates = (((h @ self.weight_hh) + (x_d @ self.weight_ih)) + self.bias)
    return gates
def forward(self, h: torch.Tensor, x_d: torch.Tensor):
    gates = (((h @ self.weight_hh) + (x_d @ self.weight_ih)) + self.bias)
    return gates