评价此页

torch.nn.functional.dropout1d#

torch.nn.functional.dropout1d(input, p=0.5, training=True, inplace=False)[源码]#

随机地将整个通道(通道是1D特征图)清零。

例如,批量输入中第ii个样本的第jj个通道是输入张量的1D张量input[i,j]\text{input}[i, j]。每次前向传播时,每个通道都会根据伯努利分布的采样,以概率p独立地清零。

有关详细信息,请参阅Dropout1d

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

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

  • inplace (bool) – 如果设置为 True,则该操作将就地执行。默认值:False

返回类型:

张量