如何在mxnet中连接符号

Question

我在MXNet中有2个符号，并希望将它们连接起来。我该怎么做：

例如：a = [100,200]，b = [300,400]，我想得到

c = [100,200,300,400]

Answer 1

您可以使用“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)