img = torch.randn(3, 224, 224)
mean = torch.tensor([.485, .456, .406])
mean = mean[:, None, None]   # (3, 1, 1)
out = img - mean             # (3, 224, 224)
print(out.shape)