Rnn forward. Args: ey: TODO. z_list: TODO. c_list: TODO. z_prev: TODO. c_prev: TODO.
(self, ey, z_list, c_list, z_prev, c_prev)
| 153 | return hs_pad.new_zeros(hs_pad.size(0), self.dunits) |
| 154 | |
| 155 | def rnn_forward(self, ey, z_list, c_list, z_prev, c_prev): |
| 156 | """Rnn forward. |
| 157 | |
| 158 | Args: |
| 159 | ey: TODO. |
| 160 | z_list: TODO. |
| 161 | c_list: TODO. |
| 162 | z_prev: TODO. |
| 163 | c_prev: TODO. |
| 164 | """ |
| 165 | if self.dtype == "lstm": |
| 166 | z_list[0], c_list[0] = self.decoder[0](ey, (z_prev[0], c_prev[0])) |
| 167 | for i in six.moves.range(1, self.dlayers): |
| 168 | z_list[i], c_list[i] = self.decoder[i]( |
| 169 | self.dropout_dec[i - 1](z_list[i - 1]), (z_prev[i], c_prev[i]) |
| 170 | ) |
| 171 | else: |
| 172 | z_list[0] = self.decoder[0](ey, z_prev[0]) |
| 173 | for i in six.moves.range(1, self.dlayers): |
| 174 | z_list[i] = self.decoder[i](self.dropout_dec[i - 1](z_list[i - 1]), z_prev[i]) |
| 175 | return z_list, c_list |
| 176 | |
| 177 | def forward(self, hs_pad, hlens, ys_pad, strm_idx=0, lang_ids=None): |
| 178 | """Decoder forward |
no outgoing calls
no test coverage detected