[256, 1024, 1024]