评价此页

torch.nn.functional.dropout#

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

在训练期间,以概率 p 随机将输入张量的某些元素置零。

使用伯努利分布的样本。

有关详细信息,请参阅 Dropout

参数
  • p (float) – 元素被置零的概率。默认为:0.5

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

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

返回类型

张量