espnet2.spk.encoder.ska_tdnn_encoder.cwSKAttention

source

class espnet2.spk.encoder.ska_tdnn_encoder.cwSKAttention(freq=40, channel=128, kernels=[3, 5], receptive=[3, 5], dilations=[1, 1], reduction=8, groups=1, L=16)

Bases: Module

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

forward(x)

Forward Function.

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