import torch
import torch.nn as nn

conv = nn.Conv2d(3, 16, kernel_size=3)
sample = torch.randn(3, 32, 32)      # one image — 3 dims

print(conv(sample).shape)            # torch.Size([16, 30, 30]) — no error