gpt4 book ai didi

python - MXNet 打印中间符号值

转载 作者:太空狗 更新时间:2023-10-30 00:47:59 27 4
gpt4 key购买 nike

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

最佳答案

环顾四周后,我发现您可以通过以下方式做到这一点:

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)

关于python - MXNet 打印中间符号值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43009881/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com