mxnet symbol 分割

Preface

1. 目的是要将一个完整的symbol在指定Op处进行分割，分别返回两段的符号
2. 昨晚一看json的关系似乎不好确立，就打算留着假期里面找时间，结果今天发现之前的替代路径有严重问题，于是迫在眉睫...

---

Function

备忘录在此...

---

Jul 31, 2017 记
程序在见后文。

---

Validation

检验员在此...

Running Test

待完善的test脚本...

import mxnet as mx
import numpy as np
import json
from sym_split import splitbyname
a= mx.sym.Variable('x')
b=a+1
y= mx.sym.Variable('y')
c=b**2+y
d= mx.sym.Convolution(c,kernel=(3,3),num_filter=3,no_bias=True,name='conv1')
d=mx.sym.Convolution(d,kernel=(4,4),num_filter=5,name='conv2')

conf = json.loads(d.tojson())
nodes = conf['nodes']

new_sym = splitbyname(d,'conv1')
new_sym
#(<Symbol feat_1>, <Symbol conv2>, [u'_plus0'], ['feat_0'])

Computing Test

上段测试程序是昨晚写的，今天没有退路了于是写完function后进行详细测试(（⊙﹏⊙）ｂ)
已经通过数值计算测试，具体程序没什么说的，贴个结果在这公示一下

>>> mod.get_outputs()[0].asnumpy()
array([[  9.99217987e-01,   7.81994022e-04]], dtype=float32)
>>> mea_mod.get_outputs()[0].asnumpy()
array([[  9.99217987e-01,   7.81994022e-04]], dtype=float32)

---

Note

1. Node赋值的时候要注意,　你操作的可能是指针?...(根据debug的结果推测，待续)
2. heads一个元素里面的3个值没有明白意义，只是猜测首个应该指向输出，剩下的两个不知所踪，多个元素生成的symbol会以group的形式输出
3. 今趟真是死里逃生... :)

---

Appendix

Jul 31, 2017 记

1. 程序在这里

---

转载于:https://www.cnblogs.com/chenyliang/p/7021015.html