Method dropout_complex

OSUM/wenet/transformer/embedding.py:253–259

(self, x)

        return pos_emb

    def dropout_complex(self, x):
        mask = torch.nn.functional.dropout(
            torch.ones_like(x.real),
            training=self.training,
            p=self.dropout_rate,
        )
        return x * mask