pytorch归一化函数

在 PyTorch 中，可以使用 `torch.nn` 模块中的 `BatchNorm1d`、`BatchNorm2d`、`BatchNorm3d`、`InstanceNorm1d`、`InstanceNorm2d`、`InstanceNorm3d` 等函数来进行归一化操作，具体使用方法如下：

import torch.nn as nn

# Batch normalization
bn = nn.BatchNorm2d(num_features=channels)

# Instance normalization
in = nn.InstanceNorm2d(num_features=channels)

其中，`num_features` 表示输入数据的特征数（即通道数）。在训练过程中，可以通过调用 `bn.train()` 和 `in.train()` 开启 BatchNormalization 和 InstanceNormalization 的训练模式，在测试过程中则可以通过调用 `bn.eval()` 和 `in.eval()` 开启测试模式。另外，还可以通过设置 `momentum`、`eps` 等参数来调整归一化的效果。