快捷方式

FrozenBatchNorm2d

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

BatchNorm2d,其中批处理统计信息和仿射参数是固定的

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

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

forward(x: Tensor) Tensor[源代码]

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

应由所有子类覆盖。

注意

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取适合初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源