espnet.nets.pytorch_backend.wavenet.CausalConv1d

source

class espnet.nets.pytorch_backend.wavenet.CausalConv1d(in_channels, out_channels, kernel_size, dilation=1, bias=True)

Bases: Module

1D dilated causal convolution.

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

forward(x)

Calculate forward propagation.

• Parameters:x (Tensor) – Input tensor with the shape (B, in_channels, T).
• Returns: Tensor with the shape (B, out_channels, T)
• Return type: Tensor