![]() |
Network utilities.
Classes
class BatchSquash
: Facilitates flattening and unflattening batch dims of a tensor.
Functions
check_single_floating_network_output(...)
maybe_permanent_dropout(...)
: Adds a Keras dropout layer with the option of applying it at inference.
mlp_layers(...)
: Generates conv and fc layers to encode into a hidden state.