无意中发现这个仓库中有绘制2d_feature代码 https://github.com/amirhfarzaneh/lsoftmax-pytorch/blob/master/train_mnist.py