Exemplo n.º 1
0
  def testGruStateTuples(self):
    self.states = recurrent_networks.gru_state_tuples(13, 'blah')
    self.RunTensor(
        self.input.sequence_gru(13, name='blah')[-1])

    for state in self.states:
      self.assertTrue(
          state[0] in self.sequence.requested_tensors, '%s missing: %s' %
          (state[0], list(six.iterkeys(self.sequence.requested_tensors))))
    self.assertEqual(
        len(self.states), len(self.sequence.requested_tensors),
        'Wrong number of Tensor states.')
Exemplo n.º 2
0
    def testGruStateTuples(self):
        self.states = recurrent_networks.gru_state_tuples(13, 'blah')
        self.RunTensor(self.input.sequence_gru(13, name='blah')[-1])

        for state in self.states:
            self.assertTrue(
                state[0] in self.sequence.requested_tensors, '%s missing: %s' %
                (state[0], list(six.iterkeys(
                    self.sequence.requested_tensors))))
        self.assertEqual(len(self.states),
                         len(self.sequence.requested_tensors),
                         'Wrong number of Tensor states.')