espnet2.spk.encoder.ska_tdnn_encoder.fwSKAttention

source

class espnet2.spk.encoder.ska_tdnn_encoder.fwSKAttention(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] Fues: [B, C, F, T] Attention weight: [K, B, 1, F, 1] Output: [B, C, F, T]