def forward()

in src/speech_reps/models/decoar2.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

    def forward(self, features, padding_mask=None):
        features = self.post_extract_proj(features)

        x = self.dropout_input(features)

        if padding_mask is not None:
            extra = padding_mask.size(1) % features.size(1)
            if extra > 0:
                padding_mask = padding_mask[:, :-extra]
            padding_mask = padding_mask.view(padding_mask.size(0), features.size(1), -1)
            padding_mask = padding_mask.all(-1)

        x = self.encoder(x, padding_mask=padding_mask)

        return x