class EmptyRatio(nn_tilde.Module):
    def __init__(self, in_channels=1, in_ratio=1, out_channels=1, out_ratio=1):
        super().__init__()

        self.out_channels = out_channels
        self.out_ratio = out_ratio
        self.in_ratio = in_ratio

        self.register_method(
            "forward",
            in_channels=in_channels, 
            in_ratio=in_ratio, 
            out_channels=out_channels, 
            out_ratio=out_ratio,
            test_buffer_size=int(8192/out_ratio*in_ratio)
        )

    @torch.jit.export
    def forward(self, x):
        b, _, l = x.shape
        return torch.zeros(b, self.out_channels, int(l / self.out_ratio * self.in_ratio))

EmptyRatio().export_to_ts('emptyratio.ts')
...
nn~ emptyratio

EmptyRatio(in_ratio=8).export_to_ts('emptyratio8.ts')
...
nn~ emptyratio8

EmptyRatio(in_ratio=5).export_to_ts('emptyratio5.ts')
...
nn~ emptyratio5