def testLstmStateTuples(self): self.states = recurrent_networks.lstm_state_tuples(13, 'blah') self.RunTensor( self.input.sequence_lstm(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.')
def testLstmStateTuples(self): self.states = recurrent_networks.lstm_state_tuples(13, 'blah') self.RunTensor(self.input.sequence_lstm(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.')