mxnet 中Symbol API 总结《2》

上次说到一些基本的Symbol的运算和在神经网络中常用的activation 和 softmax。

mxnet 中Symbol API 总结《2》

这两个的作用是根据已知的一些变量的类型来推断出变量和输出的类型。

Symbol.bind


比如刚才见到的:

a = mx.sym.Variable('a')
b = mx.sym.Variable('b')
c = a + b

ex = c.bind(ctx=mx.cpu(), args={'a' : mx.nd.ones([2,3]), 'b' : mx.nd.ones([2,3])})
ex.forward()

ex.outputs[0].asnumpy()

其中ctx就是指要用gpu还是cpu, args就是参数, 相当于开始喂数据。用字典的形式来指定。