probly.traverse_nn.utils

Generic utils for traverse_nn.

Functions

get_output_dim

Return the output feature dimension of a model.