Shortcuts

torch.nn.functional.dropout

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

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

使用来自伯努利分布的样本。

详情请参见 Dropout

Parameters
  • p (float) – 元素被置零的概率。默认值:0.5

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

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

Return type

张量

优云智算