我该如何找到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() 方法一样。我该如何找到MXNet符号中保存的实际数值。
假设我有以下代码:
x = mx.sym.Variable('x')
y = mx.sym.Variable('y')
z = x + y,
z=[400,600]
,就像 TensorFlow 的 eval() 方法一样。在稍微查找了一下之后,我发现可以通过以下方式实现:
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)