一只有恒心的小菜鸟

摘要： 继承CustomOp 定义操作符，重写前向后向方法，此时可以通过_init__ 方法传递需要用到的参数 1 class LossLayer(mxnet.operator.CustomOp): 2 def __init__(self, *args, **kwargs): 3 super(LossLay 阅读全文