个性摩托

文章
6
资源
0
加入时间
3年0月9天

MXNet创建新的操作层(详细)

本文主要介绍如何利用MXNet创建新的操作层。本文主要参考MXNet官网关于构建softmax层的例子[1],以及使用numpy定制新操作[2]两个部分。注意,这里的softmax层是指softmax损失层,也就是实际上是指softmax操作和交叉熵损失函数共同组成的层。第一部分中,我们介绍创建softmax损失层;第二部分,我们将这个损失层用到mnist分类中;第三部分,我们介绍sigmoid...