def setUp(self):
    torch.manual_seed(42)
    self.x_j = torch.randn(self.batch_size, self.input_dim)
    self.s_prev_hor = torch.randn(self.batch_size, self.cell_state_dim)
    self.s_prev_ver = torch.randn(self.batch_size, self.cell_state_dim)
    self.c_prev_hor = torch.randn(self.batch_size, self.cell_state_dim)
    self.c_prev_ver = torch.randn(self.batch_size, self.cell_state_dim)
    self.device = torch.device('cpu')
def setUp(self):
    torch.manual_seed(42)
    self.x_j = torch.randn(self.batch_size, self.input_dim)
    self.s_prev_hor = torch.randn(self.batch_size, self.cell_state_dim)
    self.s_prev_ver = torch.randn(self.batch_size, self.cell_state_dim)
    self.c_prev_hor = torch.randn(self.batch_size, self.cell_state_dim)
    self.c_prev_ver = torch.randn(self.batch_size, self.cell_state_dim)
    self.device = torch.device('cpu')