def decode(self, enc): enc = enc.repeat(self.out_length, 1, 1) h_dec, _ = self.dec_lstm(enc) h_dec = h_dec.permute(1, 0, 2) fut_pred = self.op(h_dec) fut_pred = fut_pred.permute(1, 0, 2) fut_pred = outputActivation(fut_pred) return fut_pred
def decode(self, enc): enc = enc.repeat(self.out_length, 1, 1) h_dec, _ = self.dec_lstm(enc) # if h_dec.shape[1]==self.decoder_size: # h_dec = self.bn_dec(h_dec) h_dec = h_dec.permute(1, 0, 2) fut_pred = self.op(h_dec) fut_pred = self.bn_lin(fut_pred) fut_pred = fut_pred.permute(1, 0, 2) fut_pred = self.dropout(fut_pred) fut_pred = outputActivation(fut_pred) return fut_pred