深度学习 .concat

在 MXNet 中，.concat() 方法用于沿指定轴将多个数组连接在一起。它可以对一维、二维或更高维的 ndarray 进行操作。

语法

mx.nd.concat(*arrays, dim=0)

参数

• arrays: 要连接的多个 ndarray 对象。
• dim: 指定沿哪个轴进行连接。默认为 0（沿第一个轴连接）。

返回值

• 返回一个新的 ndarray，其形状取决于输入数组的形状和连接的轴。

示例

import mxnet as mx

# 创建两个 ndarray
arr1 = mx.nd.array([[1, 2], [3, 4]])
arr2 = mx.nd.array([[5, 6]])

# 沿第一个轴 (行) 连接
result1 = mx.nd.concat(arr1, arr2, dim=0)

print(result1)
# 输出:
# [[1. 2.]
#  [3. 4.]
#  [5. 6.]]

# 沿第二个轴 (列) 连接
arr3 = mx.nd.array([[7], [8]])
result2 = mx.nd.concat(arr1, arr3, dim=1)

print(result2)
# 输出:
# [[1. 2. 7.]
#  [3. 4. 8.]]

注意事项

• 所有输入数组在连接的轴上必须具有相同的形状。
• 如果输入数组的维度不一致，连接操作将引发错误。

应用场景

• .concat() 方法通常用于数据预处理、特征工程以及模型输入的准备等场景。