quantlab.model.predefined.tcn.Chomp1d#

class quantlab.model.predefined.tcn.Chomp1d(chomp_size)[source]#

Bases: Module

Drops the last chomp_size steps, the right padding of a causal convolution.

forward(x)[source]#

Return x[:, :, :-chomp_size].

Examples

>>> Chomp1d(2)(torch.zeros(1, 3, 5)).shape
torch.Size([1, 3, 3])