speechbrain.nnet.dropout 模块

实现dropout的库。

Authors

• Mirco Ravanelli 2020

摘要

类：

Dropout2d

此函数实现了2d dropout。

参考

class speechbrain.nnet.dropout.Dropout2d(drop_rate, inplace=False)

基础：Module

此函数实现了dropout 2d。它会随机将整个通道置为零。

Parameters:

• drop_rate (float) – 这是丢弃因子（介于0和1之间）。

• inplace (bool) – 如果为True，则使用原地操作。

Example

>>> drop = Dropout2d(drop_rate=0.5)
>>> inputs = torch.rand(10, 50, 40)
>>> output=drop(inputs)
>>> output.shape
torch.Size([10, 50, 40])

forward(x)

对输入张量应用2d dropout。

Parameters:

x (torch.Tensor (batch, time, channel1, channel2)) – 输入以进行归一化。期望为4维张量。

Returns:

x_drop – 被零化的张量。

Return type:

torch.Tensor