from .sort import Split, SortLayer

from .layers import SparseLayer, NASLayer, Convolution, transform_means, transform_sigmas
from .layers import ngenerate, transform_means, densities

from .tensors import contract, logsoftmax, batchmm

# from .tensors import flatten_indices_mat

from .sort import Split, SortLayer

from .layers import SparseLayer, NASLayer, Convolution, transform_means, transform_sigmas
from .layers import ngenerate, transform_means, densities

from .tensors import contract, logsoftmax, batchmm, simple_normalize

# from .tensors import flatten_indices_mat

Fix bug