mean = torch.tensor([0.485, 0.456, 0.406])   # (3,)
img = torch.randn(3, 224, 224)               # (C, H, W)

img - mean