2017-03-24 6 views
6

MXNetに2つのシンボルがあり、それらを連結したいと思います。mxnetでシンボルを連結する方法

例えば:私はこの方法を行うことができますa = [100,200]b = [300,400]を、IDは、 "Concat" メソッドを使用してこれを行うことができ

c = [100,200,300,400]

答えて

7

を取得したいです。

a = mx.sym.Variable('a') 
b = mx.sym.Variable('b') 
c = mx.sym.Concat(a,b,dim=0) 

これを確認するには、あなたがチェックするためにexecutorを使用してシンボルを実行することができます。

e = c.bind(mx.cpu(), {'a': mx.nd.array([100,200]), 'b':mx.nd.array([300,400])}) 
y = e.forward() 
y[0].asnumpy() 

あなたは出力が得られます:

array([ 100., 200., 300., 400.], dtype=float32) 
関連する問題