Method __init__

modules/sequence_modeling.py:41–46

(self, input_size, hidden_size, output_size)

class BidirectionalLSTM(nn.Module):

    def __init__(self, input_size, hidden_size, output_size):
        super(BidirectionalLSTM, self).__init__()
        self.rnn = nn.LSTM(input_size, hidden_size, bidirectional=True, batch_first=True)
        self.linear = nn.Linear(hidden_size * 2, output_size)
        # self.h0 = torch.randn(2, 1, hidden_size).cuda()
        # self.c0 = torch.randn(2, 1, hidden_size).cuda()

    def forward(self, input):
        """