The output shape of PSPNet is `(batch, class_num, h, w)`. Now i want to get the feature(one vector) of every class. What shuold i do to get it? Thanks!