espnet2.spk.encoder.ska_tdnn_encoder.SKAttentionModule

source

class espnet2.spk.encoder.ska_tdnn_encoder.SKAttentionModule(channel=128, reduction=4, L=16, num_kernels=2)

Bases: Module

Initializes internal Module state, shared by both nn.Module and ScriptModule.

forward(x, convs)

Forward function.

Input: [B, C, T] Split: [K, B, C, T] Fues: [B, C, T] Attention weight: [B, C, 1] Output: [B, C, T]