torch.nn.functional.dropout

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

在训练过程中，以概率 p 随机将输入张量的一些元素置为零。

使用伯努利分布的样本。

详细内容请见 Dropout 。

参数:

• p (浮点数) – 元素被清零的概率。默认值：0.5

• 训练（布尔值）- 如果是 True 则应用 dropout。默认： True

• inplace（布尔值）- 如果设置为 True ，将在此处执行此操作。默认： False

返回类型:

张量