PyTorch provides 2 kinds of Softmax class. The one is applying softmax along a certain dimension. The other is do softmax on a spatial matrix sized in B, C, H, W.
But it seems like some problems existing in Softmax2d. : (
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:PyTorch Softmax - Python技术站