0
我有一个关于在mxnet中创建自定义激活功能/ op使用的语法的问题。我一直在寻找这个例子: https://github.com/dmlc/mxnet/blob/master/example/numpy-ops/custom_softmax.pymxnet自定义激活功能/ op在numpy
具体来说,这一部分:
class Softmax(mx.operator.CustomOp):
def forward(self, is_train, req, in_data, out_data, aux):
x = in_data[0].asnumpy()
y = np.exp(x - x.max(axis=1).reshape((x.shape[0], 1)))
y /= y.sum(axis=1).reshape((x.shape[0], 1))
self.assign(out_data[0], req[0], mx.nd.array(y))
def backward(self, req, out_grad, in_data, out_data, in_grad, aux):
l = in_data[1].asnumpy().ravel().astype(np.int)
y = out_data[0].asnumpy()
y[np.arange(l.shape[0]), l] -= 1.0
self.assign(in_grad[0], req[0], mx.nd.array(y))
怎么了IN_DATA [0] VS IN_DATA [1],OUT_DATA [0] VS OUT_DATA [1]?这些指数对应什么?
谢谢!