将矩阵A: m×n×p 转化为 m×n×p×1 以实现多维矩阵叠加的效果:
import numpy as np
A = np.array([ A ]) // A为待操作矩阵
这样得到的是(1,m,n,p)
如果想得到(m,n,p,1)的维度矩阵,则:
A =A [:, : ,:,np.newaxis]
将矩阵A: m×n×p 转化为 m×n×p×1 以实现多维矩阵叠加的效果:
import numpy as np
A = np.array([ A ]) // A为待操作矩阵
这样得到的是(1,m,n,p)
如果想得到(m,n,p,1)的维度矩阵,则:
A =A [:, : ,:,np.newaxis]