2017-03-24 8 views
5

W jaki sposób znaleźć rzeczywiste wartości liczbowe przechowywane w symbolu MXNet.Wartości symboli pośrednich drukowania MXNet

Załóżmy, że mam,

x = mx.sym.Variable('x') 
y = mx.sym.Variable('y') 
z = x + y, 

jeśli x = [100200] oraz y = [300400] chcę wydrukować:

z = [400,600],

rodzaj eval podoba tensorflow za () metoda

Odpowiedz

8

Po obejrzeniu trochę, znalazłem, że możesz to zrobić przez:

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

daje 'wyjście':

[<NDArray 2 @cpu(0)>] 

aby wydrukować rzeczywiste wyjście liczbowym:

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