Shortcuts

FrozenBatchNorm2d

class torchvision.ops.FrozenBatchNorm2d(num_features: int, eps: float = 1e-05)[source]

BatchNorm2d,其中批次统计数据和仿射参数是固定的

Parameters:
  • num_features (int) – 特征数量 C 来自预期输入的大小 (N, C, H, W)

  • eps (float) – 为了数值稳定性而添加到分母的值。默认值:1e-5

forward(x: Tensor) Tensor[source]

定义每次调用时执行的计算。

应该由所有子类覆盖。

注意

尽管前向传递的配方需要在此函数内定义,但之后应该调用Module实例而不是这个,因为前者负责运行已注册的钩子,而后者则默默地忽略它们。