iamhankai / cpp-pytorch

A bite of cpp api in PyTorch-1.0
165 stars 40 forks source link

how can i use forward() when i have Multiple inputs and outputs #16

Closed smallsheep666 closed 5 years ago

smallsheep666 commented 5 years ago

In my python code my forward is defined as: def forward(self, input_seq, hidden): output_seq, hidden = self.gru(input_seq, hidden) if isinstance(output_seq, torch.nn.utils.rnn.PackedSequence): outputseq, = torch.nn.utils.rnn.pad_packed_sequence( output_seq, batch_first=False) mean = self.linear_mean2(F.relu(self.linear_mean1(output_seq))) return mean, hidden how can i use forward() in c++ ?