neuralmonkey.nn.utils module

Module which provides utility functions used across the package.

neuralmonkey.nn.utils.dropout(variable: tensorflow.python.framework.ops.Tensor, keep_prob: float, train_mode: tensorflow.python.framework.ops.Tensor) → tensorflow.python.framework.ops.Tensor

Perform dropout on a variable, depending on mode.

Parameters:

• variable – The variable to be dropped out
• keep_prob – The probability of keeping a value in the variable
• train_mode – A bool Tensor specifying whether to dropout or not