TensorRT softmax层
今天遇到分类问题TensorRT softmax层如果直接在fc后面接softmax 则会进行全局softmax需要进行reshape 比如2分类 需要转为[1, 1, 2]这种维度 然后在维度2上今天softmax百度后 先使用IShuffleLayer进行维度变换IShuffleLayer *shuffleLayer = network->addShuffle(input);assert(shuffleLayer);shuffleLayer->setR...
