espnet2.diar.layers.tcn_nomask.TemporalConvNet

source

class espnet2.diar.layers.tcn_nomask.TemporalConvNet(N, B, H, P, X, R, norm_type='gLN', causal=False)

Bases: Module

Basic Module of tasnet.

• Parameters:
  • N – Number of filters in autoencoder
  • B – Number of channels in bottleneck 1 * 1-conv block
  • H – Number of channels in convolutional blocks
  • P – Kernel size in convolutional blocks
  • X – Number of convolutional blocks in each repeat
  • R – Number of repeats
  • norm_type – BN, gLN, cLN
  • causal – causal or non-causal

forward(mixture_w)

Keep this API same with TasNet.

• Parameters:mixture_w – [M, N, K], M is batch size
• Returns: [M, B, K]
• Return type: bottleneck_feature