ncdl.nn.LatticeBatchNorm

class ncdl.nn.LatticeBatchNorm(*args: Any, **kwargs: Any)
__init__(lattice: Lattice, num_features: int, eps: float = 1e-05, momentum: float = 0.1, affine: bool = True, track_running_stats: bool = True, device=None, dtype=None) None

Methods

__init__(lattice, num_features[, eps, ...])

forward(input)