speechbrain.nnet.dropout 模块

实现 dropout 的库。

作者

• Mirco Ravanelli 2020

摘要

类

Dropout2d

此函数实现 dropout 2d。

参考

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

基类: Module

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

参数:

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

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

示例

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

forward(x)

将 dropout 2d 应用于输入张量。

参数:

x (torch.Tensor (batch, time, channel1, channel2)) – 要进行 dropout 的输入。期望为 4D 张量。

返回:

x_drop – 通道被清零的张量。

返回类型:

torch.Tensor