torch.nn.functional.dropout

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

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

使用伯努利分布的样本。

详情请参阅 Dropout。

参数:

• p (float) – 元素被归零的概率。默认值：0.5

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

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

返回类型:

张量