快捷方式

FrozenBatchNorm2d

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

BatchNorm2d,其中批统计量和仿射参数已固定

参数:
  • num_features (int) – 对于预期大小为 (N, C, H, W) 的输入的特征 C 的数量

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

forward(x: Tensor) Tensor[来源]

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

所有子类都应重写此方法。

注意

尽管前向传播的实现需要在此函数中定义,但您应该在之后调用 Module 实例而不是此函数,因为前者会处理注册的钩子,而后者则会静默忽略它们。

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源