probly.method.het_nets.torch

Torch implementation of HetNets.

Functions

drop_in_place_het_layer	Replace the last linear layer with a HeteroscedasticLayer.
remove_layer	Remove the softmax layer.
skip_layer	Traverser for torch HetNets.