如何在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)