当前位置:  开发笔记 > 编程语言 > 正文

MXNet打印中间符号值

如何解决《MXNet打印中间符号值》经验,为你挑选了1个好方法。

如何找到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()方法



1> Karishma Mal..:

环顾四周后,我发现你可以这样做:

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()

会给你'输出':

[]

要打印实际的数字输出:

print output[0].asnumpy()
array([ 400.,  600.], dtype=float32)

推荐阅读
重庆制造漫画社
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有