CNTK python中的用户定义层

Question

我尝试使用python接口创建一个在CNTK中使用的自定义层。我正在跟踪this guide，但在类的__init__中仍有一个TypeError异常抛出。请注意，我只是复制粘贴了链接指南中的示例。在

import cntk as C
import numpy as np

class MySigmoid(UserFunction):
    def __init__(self, arg, name='MySigmoid'):
        super(MySigmoid, self).__init__([arg], name=name)

    def forward(self, argument, device=None, outputs_to_retain=None):
        sigmoid_x = 1 / (1 + np.exp(-argument))
        return sigmoid_x, sigmoid_x

    def backward(self, state, root_gradients):
        sigmoid_x = state
        return root_gradients * sigmoid_x * (1 - sigmoid_x)

    def infer_outputs(self):
        return [output_variable(self.inputs[0].shape, self.inputs[0].dtype,
            self.inputs[0].dynamic_axes)]

    @staticmethod
    def deserialize(inputs, name, state):
        return MySigmoid(inputs[0], name)
model = C.layers.Sequential(C.layers.Dense(10), C.user_function(layers_extensions.MySigmoid(3)))

我得到的错误是：

^{pr2}$

我试图用谷歌搜索这个错误，但是没有结果。你能帮助我吗？在

Answer 1

出于某些原因，CNTK将argument方法中的argument参数作为列表传递，即使它是单个参数。最后我从名单上取了第一个。您将找到工作示例here。在