[128, 512, 512]