寒武纪实现高维向量的softmax进阶优化和库函数对比

admin2024-07-01  17

关于寒武纪编程可以参考本人之前的文章添加链接描述,添加链接描述,添加链接描述
寒武纪实现高维向量的softmax进阶优化和库函数对比,在这里插入图片描述,第1张
实验证明,axis=0和axis=-1的时候,手写softmax速度可以和库函数媲美,甚至于更甚一筹。

src/softmax.mlu

#include <bang.h>
#include 
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明原文出处。如若内容造成侵权/违法违规/事实不符,请联系SD编程学习网:675289112@qq.com进行投诉反馈,一经查实,立即删除!