【问题标题】:MXNet print intermediate symbol valuesMXNet 打印中间符号值
【发布时间】:2017-03-24 22:03:36
【问题描述】:
如何找到 MXNet 符号中保存的实际数值。
假设我有,
x = mx.sym.Variable('x')
y = mx.sym.Variable('y')
z = x + y,
如果 x = [100,200] 且 y=[300,400],
我要打印:
z = [400,600],
有点像 tensorflow 的 eval() 方法
【问题讨论】:
标签:
python
machine-learning
deep-learning
mxnet
【解决方案1】:
看了一圈后,我发现你可以这样做:
x = mx.sym.Variable('x')
y = mx.sym.Variable('y')
z = x + y
executor = z.bind(mx.cpu(), {'x': mx.nd.array([100,200]), 'y':mx.nd.array([300,400])})
output = executor.forward()
会给你“输出”:
[<NDArray 2 @cpu(0)>]
要打印实际的数值输出:
print output[0].asnumpy()
array([ 400., 600.], dtype=float32)