Reshape label of VectorLabelLayer to be (batch, label_size, 1, 1) rather than (batch, 1, label_size, 1)