model = nn.Sequential(
    nn.Conv2d(3, 16, kernel_size=3),
    nn.BatchNorm2d(16),
)

model(sample)