摘要: 继承CustomOp 定义操作符,重写前向后向方法,此时可以通过_init__ 方法传递需要用到的参数 1 class LossLayer(mxnet.operator.CustomOp): 2 def __init__(self, *args, **kwargs): 3 super(LossLay 阅读全文
posted @ 2017-09-11 15:43 一只有恒心的小菜鸟 阅读(1268) 评论(0) 推荐(0) 编辑