>>> import mxnet.ndarray as nd
>>> x = [[1,1],[2,2]]
>>> y = [[3,3],[4,4],[5,5]]
>>> z = [[6,6], [7,7],[8,8]]
>>> nd.concat(x,y,z,dim=0) # 按照一维拼接,相当于竖着拼接。
[[ 1. 1.]
[ 2. 2.]
[ 3. 3.]
[ 4. 4.]
[ 5. 5.]
[ 6. 6.]
[ 7. 7.]
[ 8. 8.]]
<NDArray 8x2 @cpu(0)>
>>> concat(y,z,dim=1) # 按照二维拼接,相当于横着拼接。
[[ 3. 3. 6. 6.]
[ 4. 4. 7. 7.]
[ 5. 5. 8. 8.]]
<NDArray 3x4 @cpu(0)>
mxnet.ndarray进行矩阵拼接,使用concat命令。
最新推荐文章于 2022-08-06 23:18:36 发布