Esempio n. 1
0
    def get_initial_state(self, inputs=None, batch_size=None, dtype=None):
        initial_states = []
        for cell in self.cells[::
                               -1] if self.reverse_state_order else self.cells:
            get_initial_state_fn = getattr(cell, 'get_initial_state', None)
            if get_initial_state_fn:
                initial_states.append(
                    get_initial_state_fn(inputs=inputs,
                                         batch_size=batch_size,
                                         dtype=dtype))
            else:
                initial_states.append(
                    rnn_utils.generate_zero_filled_state_for_cell(
                        cell, inputs, batch_size, dtype))

        return tuple(initial_states)
Esempio n. 2
0
 def get_initial_state(self, inputs=None, batch_size=None, dtype=None):
     return rnn_utils.generate_zero_filled_state_for_cell(
         self, inputs, batch_size, dtype)