Perform dropout. Args: input_tensor: float Tensor. dropout_prob: Python float. The probability of dropping out a value (NOT of *keeping* a dimension as in `tf.nn.dropout`). Returns: A version of `input_tensor` with dropout applied.
(input_tensor, dropout_prob)
| 126 | |
| 127 | |
| 128 | def dropout(input_tensor, dropout_prob): |
| 129 | """Perform dropout. |
| 130 | |
| 131 | Args: |
| 132 | input_tensor: float Tensor. |
| 133 | dropout_prob: Python float. The probability of dropping out a value (NOT of |
| 134 | *keeping* a dimension as in `tf.nn.dropout`). |
| 135 | |
| 136 | Returns: |
| 137 | A version of `input_tensor` with dropout applied. |
| 138 | """ |
| 139 | if dropout_prob is None or dropout_prob == 0.0: |
| 140 | return input_tensor |
| 141 | output = tf.nn.dropout(input_tensor, rate=dropout_prob) |
| 142 | return output |
| 143 | |
| 144 | |
| 145 | def get_attention_mask(nd, ns, *, dtype): |
no outgoing calls
no test coverage detected