Also, I just tried:
        self.conv1 = torch.nn.Conv1d(
            in_channels=feature_dim,
            out_channels= feature_dim,
            kernel_size=kernel_sizes[0],
            stride=1,
            groups= feature_dim,
            padding=kernel_sizes[0] // 2)

This still doesn’t quite get it:
inp tensor(3.2217e-05) tensor(0.1359) tensor(1.9529) torch.Size([1, 161, 32])

out1 tensor(2.0011e-07, grad_fn=<MinBackward1>) tensor(0.0008, grad_fn=<MeanBackward0>) tensor(0.0121, grad_fn=<MaxBackward1>) torch.Size([1, 161, 32])