【问题标题】: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)
    

    【讨论】:

      猜你喜欢
      • 2017-06-01
      • 1970-01-01
      • 1970-01-01
      • 2017-03-25
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多