Torchélie
latest
Pytorch Utils
torchelie.nn
Convolutions
Normalization
Misc
VQ
MultiVQ
Noise
Debug
Dummy
Lambda
Reshape
Interpolate2d
InterpolateBilinear2d
AdaptiveConcatPool2d
ModulatedConv
SelfAttention2d
GaussianPriorFunc
UnitGaussianPrior
InformationBottleneck
Const
SinePositionEncoding2d
MinibatchStddev
Blocks
Sequential
Activations
torchelie.nn.utils
torchelie.optim
torchelie.lr_scheduler
torchelie.utils
torchelie.data_learning
torchelie.loss
torchelie.models
torchelie.distributions
torchelie.datasets
torchelie.transforms
torchelie.recipes
Algorithms and training
Understanding recipes
Using the predefined recipes and callbacks
torchelie.callbacks
torchelie.hyper
Torchélie
Docs
»
torchelie.nn
»
InformationBottleneck
Edit on GitHub
InformationBottleneck
¶
class
torchelie.nn.
InformationBottleneck
(
in_channels
,
num_latents
,
strength
=
1
,
kl_reduction
=
'mean'
)
¶
training
:
bool
¶
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.