评价此页

torch.nn.functional.dropout3d#

torch.nn.functional.dropout3d(input, p=0.5, training=True, inplace=False)[source]#

随机将整个通道(一个3D特征图)设置为零。

例如,在批量输入中,第 ii 个样本的第 jj 个通道是一个3D张量 input[i,j]\text{input}[i, j]。每个通道将在每次前向调用时,根据 p 的概率,使用伯努利分布的样本独立地被置零。

请参阅 Dropout3d 以获取详细信息。

参数
  • p (float) – 通道被置零的概率。默认为 0.5。

  • training (bool) – 如果为 True,则应用 dropout。默认为 True

  • inplace (bool) – 如果设置为 True,则执行此操作的原地修改。默认为 False

返回类型

张量