You could permute the activation and pass the right shape via:
print(out.shape) # [batch_size, out_channels, seq_len]
out = out.permute(0, 2, 1) # [batch_size, seq_len, out_channels]

This would shift the temporal dimension to dim1 and the output channels to dim2.