快捷方式

BNReLU3d

class torch.ao.nn.intrinsic.BNReLU3d(batch_norm, relu)[source][source]

这是一个按顺序调用 BatchNorm 3d 和 ReLU 模块的容器。在量化期间,它将被相应的融合模块替换。

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并解答疑问

查看资源