def test_dimensions(self):
    '\n        Tests if the input and output dimensions of the cell are as expected.\n        '
    cell = LSTM2dCell(self.input_dim, self.cell_state_dim, self.device)
    (c_ji, s_ji) = cell.forward(x=self.x_j, s_prev_hor=self.s_prev_hor, s_prev_ver=self.s_prev_ver, c_prev_hor=self.c_prev_hor, c_prev_ver=self.c_prev_ver)
    c_shape = list(c_ji.shape)
    s_shape = list(s_ji.shape)
    self.assertEqual(c_shape, [self.batch_size, self.cell_state_dim], 'Next cell state has unexpected shape')
    self.assertEqual(s_shape, [self.batch_size, self.cell_state_dim], 'Next hidden state has unexpected shape')
Tests if the input and output dimensions of the cell are as expected.
def test_dimensions(self):
    '\n        \n        '
    cell = LSTM2dCell(self.input_dim, self.cell_state_dim, self.device)
    (c_ji, s_ji) = cell.forward(x=self.x_j, s_prev_hor=self.s_prev_hor, s_prev_ver=self.s_prev_ver, c_prev_hor=self.c_prev_hor, c_prev_ver=self.c_prev_ver)
    c_shape = list(c_ji.shape)
    s_shape = list(s_ji.shape)
    self.assertEqual(c_shape, [self.batch_size, self.cell_state_dim], 'Next cell state has unexpected shape')
    self.assertEqual(s_shape, [self.batch_size, self.cell_state_dim], 'Next hidden state has unexpected shape')