torch.nn.functional.dropout3d

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

随机将整个通道置零（一个通道是一张三维特征图）。

例如，批处理输入中的第$i$个样本的第$j$个通道是一个3D张量$\text{input}[i, j]$，它是输入张量的一部分。每个通道将在每次前向调用时以概率p独立地被置零，该概率是从伯努利分布中采样得到的。

详情请参阅Dropout3d。

参数

• p (float) – 指定频道被置为零的概率，默认值为 0.5。

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

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

返回类型

Tensor