GroupNorm

class GroupNorm(num_groups, num_channels, eps=1e-05, affine=True, **kwargs)

GroupNorm 的简单实现。当前仅支持四维张量。参考文献：https://arxiv.org/pdf/1803.08494.pdf。

上一页

SyncBatchNorm

下一页

InstanceNorm