torch.nn.functional.dropout

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

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

使用伯努利分布的样本。

详情请参阅Dropout。

参数

• p (float) – 控制元素被置为零的概率。默认值：0.5

• training (bool) – 是否应用 dropout，默认为 True

• inplace (bool) – 如果设置为True，将以原地方式执行此操作。默认值： False

返回类型

Tensor