我在MXNet中有两个符号,想将它们拼接起来。如何实现这一点呢:
例如:a = [100,200]
,b = [300,400]
,我想得到
c = [100,200,300,400]
回答:
你可以使用”Concat“方法来实现这一点。
a = mx.sym.Variable('a')b = mx.sym.Variable('b')c = mx.sym.Concat(a,b,dim=0)
为了验证这一点,你可以使用执行器来执行你的符号进行检查:
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)