Hello all, In caffe I used the SoftmaxWithLoss for multiple class segmentation problem

(Caffe) block (n) --> BatchNorm -> ReLU --> SoftmaxWithLoss

Which loss in pytorch should I used to achieve similar result in caffe? Thanks