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

How do you concatenate symbols in mxnet