espnet2.enh.layers.uses.ChannelTAC

source

class espnet2.enh.layers.uses.ChannelTAC(input_dim, eps=1e-05)

Bases: Module

Channel Transform-Average-Concatenate (TAC) module.

• Parameters:
  • input_dim (int) – dimension of the input feature.
  • eps (float) – epsilon for layer normalization.

forward(x, ref_channel=None)

ChannelTAC Forward.

• Parameters:
  • x (torch.Tensor) – input feature (batch, C, N, freq, time)
  • ref_channel (None or int) – index of the reference channel.
• Returns: output feature (batch, C, N, freq, time)
• Return type: output (torch.Tensor)