espnet2.asr.state_spaces.components.DropoutNd

source

class espnet2.asr.state_spaces.components.DropoutNd(p: float = 0.5, tie=True, transposed=True)

Bases: Module

Initialize dropout module.

tie: tie dropout mask across sequence lengths (Dropout1d/2d/3d)

forward(X)

Forward pass.

X: (batch, dim, lengths…)