torch.nn.functional.dropout3d¶
- torch.nn.functional.dropout3d(input, p=0.5, training=True, inplace=False)[source][source]¶
随机将整个通道置零(通道是一个 3D 特征图)。
例如,批处理输入中的第 个通道的第 个样本是输入张量 的 3D 张量。每个通道将在每次前向调用时独立地以概率
p
使用伯努利分布的样本进行置零。详细内容请见
Dropout3d
。- 参数:
p (浮点数) – 通道置零的概率。默认值:0.5
training (布尔值) – 如果是
True
则应用 dropout。默认值:True
inplace(布尔值)- 如果设置为
True
,将在此处执行此操作。默认:False
- 返回类型: