(self, model, output)
| 872 | return output |
| 873 | |
| 874 | def _apply_dropout(self, model, output): |
| 875 | if self.dropout_ratio and not self.forward_only: |
| 876 | with core.NameScope(self.name or ''): |
| 877 | output = brew.dropout( |
| 878 | model, |
| 879 | output, |
| 880 | str(output) + '_with_dropout_mask{}'.format(self.mask), |
| 881 | ratio=float(self.dropout_ratio), |
| 882 | is_test=self.is_test, |
| 883 | use_cudnn=self.use_cudnn, |
| 884 | ) |
| 885 | self.mask += 1 |
| 886 | return output |
| 887 | |
| 888 | |
| 889 | class MultiRNNCellInitializer: |
no test coverage detected