会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
一只有恒心的小菜鸟
博客园
首页
新随笔
联系
管理
2017年9月5日
使用python创建mxnet操作符(网络层)
摘要: 对cuda了解不多,所以使用python创建新的操作层是个不错的选择,当然这个性能不如cuda编写的代码。 在MXNET源码的example/numpy-ops/下有官方提供的使用python编写新操作符的实例。分别跑ndarray_softmax.py、numpy_softmax.py和custo
阅读全文
posted @ 2017-09-05 10:40 一只有恒心的小菜鸟
阅读(1441)
评论(0)
推荐(0)
编辑
公告