[32, 3, 224, 224]