pyprobound.layers.module

Layer wrapping arbitrary modules.

Members are explicitly re-exported in pyprobound.layers.

Classes

ModuleLayer(layer_spec, input_shape, ...)

Layer wrapper for torch.nn.Module.

ModuleSpec(alphabet, module[, name])

Specification for any torch.nn.Module.