在训练期间应用 3D 通道丢弃。
在训练期间,以概率 \(p\) 随机且独立地将整个通道归零。此层期望通道位于最后,即输入形状应为 NDHWC 或 DHWC,其中:N 是批量维度,D 是深度,H 是输入图像高度,W 是输入图像宽度,C 是输入通道数。
剩余通道将按 \(\frac{1}{1-p}\) 进行缩放,以保持每个元素的期望值。与将单个条目归零的传统丢弃不同,此层将整个通道归零。这通常有利于处理 3D 数据的卷积层,例如在医学图像处理或视频处理中。
- 参数:
p (float) – 训练期间将通道归零的概率。
方法