torch.nn.functional.dropout1d¶
- torch.nn.functional.dropout1d(input, p=0.5, training=True, inplace=False)[source][source]¶
随机置零整个通道(通道是一个 1D 特征图)。
例如,批处理输入中的第 个通道的第 个样本是输入张量的 1D 张量 。每个通道将在每次前向调用时独立地以概率
p
使用伯努利分布的样本进行置零。详细内容请见
Dropout1d
。- 参数:
p (浮点数) – 通道置零的概率。默认值:0.5
training (布尔值) – 如果是
True
则应用 dropout。默认值:True
inplace(布尔值)- 如果设置为
True
,将在此处执行此操作。默认:False
- 返回类型: