Method get_init_h

cdslib/core/nn/modules/lstm.py:244–253

(self, batch_size=1, device=torch.device("cpu"))

            assert dim_output == dim_rnn

    def get_init_h(self, batch_size=1, device=torch.device("cpu")):
        return nn_utils.get_constant_rnn_hidden_states(
            "lstm",
            self.num_rnn_layers,
            self.rnn_feature_dim,
            batch_size,
            bidirectional=self.bidirectional,
            const=0.0,
            device=device,
        )

    def forward(self, x, h0=None, valid_seq_lens=None):
        """