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，则应用丢弃。默认值：True

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

返回类型

张量