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