x3板端部署时,模型内部而非输出节点的softmax能否部署到bpu上?如果不行的话有什么好的有效算子可以替换吗

x3板端部署时,模型内部而非输出节点的softmax能否部署到bpu上?如果不行的话有什么好的有效算子可以替换吗
例如yolo里面的注意力机制,我把注意力机制做了修改,但是softmax还是需要保留,此时有什么好的softmax算子替换方法吗?

bpu 对softmax比较敏感,具体观察工具链手册是否支持算子即可