espnet2.asr.preencoder.sinc.SpatialDropout

source

class espnet2.asr.preencoder.sinc.SpatialDropout(dropout_probability: float = 0.15, shape: tuple | list | None = None)

Bases: Module

Spatial dropout module.

Apply dropout to full channels on tensors of input (B, C, D)

Initialize.

• Parameters:
  • dropout_probability – Dropout probability.
  • shape (tuple , list) – Shape of input tensors.

forward(x: Tensor) → Tensor

Forward of spatial dropout module.