摘要: 对cuda了解不多,所以使用python创建新的操作层是个不错的选择,当然这个性能不如cuda编写的代码。 在MXNET源码的example/numpy-ops/下有官方提供的使用python编写新操作符的实例。分别跑ndarray_softmax.py、numpy_softmax.py和custo 阅读全文
posted @ 2017-09-05 10:40 一只有恒心的小菜鸟 阅读(1431) 评论(0) 推荐(0) 编辑