(state_before, use_noise, trng)
| 74 | |
| 75 | |
| 76 | def dropout_layer(state_before, use_noise, trng): |
| 77 | proj = tensor.switch(use_noise, |
| 78 | (state_before * |
| 79 | trng.binomial(state_before.shape, |
| 80 | p=0.5, n=1, |
| 81 | dtype=state_before.dtype)), |
| 82 | state_before * 0.5) |
| 83 | return proj |
| 84 | |
| 85 | |
| 86 | def _p(pp, name): |