torch.nn.LazyInstanceNorm3d
class LazyInstanceNorm3d extends _LazyInstanceNormLazy Instance Normalization 3D: automatically infers channels for 3D instance normalization.
Extends InstanceNorm3d with lazy initialization. Normalizes each volumetric sample's channels independently. Useful for 3D medical imaging, volumetric style transfer, and other applications where instance normalization is preferred over batch normalization.
- Medical imaging: Useful for normalizing medical scans without batch statistics
- Per-instance: Each 3D volume normalized independently
- Generative models: Sometimes preferred over batch norm in 3D generative models
Examples
// 3D volumetric instance normalization (medical imaging)
const lazy_in = new torch.nn.LazyInstanceNorm3d();
const volume = torch.randn([4, 1, 64, 64, 64]); // [batch, channels, D, H, W]
const normalized = lazy_in.forward(volume); // Initializes with 1 channel
// Each volume normalized independently