minuet.nn.normalizations

Classes

BatchNorm(num_features[, eps, momentum, ...])

Applies torch.nn.BatchNorm1d on the feature tensor of the given SparseTensor x.

class BatchNorm(num_features: int, eps: float = 1e-5, momentum: float = 0.1, affine: bool = True, track_running_stats: bool = True, device=None, dtype=None)

Applies torch.nn.BatchNorm1d on the feature tensor of the given SparseTensor x.