def initial_state()

in monobeast/minigrid/monobeast_amigo.py [0:0]

• 4 lines of code
• 3 McCabe index (conditional complexity)

    def initial_state(self, batch_size):
        """Initializes LSTM."""
        if not self.use_lstm:
            return tuple()
        return tuple(torch.zeros(self.core.num_layers, batch_size, self.core.hidden_size) for _ in range(2))