1、首先,在Python中安装matlab engine for python,从而能够在Python中调用MATLAB。详情请见引用的经验。并在python中启动matlab engine
2、之后,对数组进行一系列的操作,如果5维数组是[16,1,512,64,64]这种,为了简化计算复杂度,可以首先用squeeze函数将数组降维为4维[16,512,512];假如矩阵是torch类型的,需要先将其变为ndarray格式再转为cpu(如果在GPU上进行处理),之后再变成列表传入MATLAB。如图所示。
3、接下来,在MATLAB中,MATLAB会将导入的list看作cell。因此需要先将cell展开成1*N的cell。
4、然后,再将1*N的cell用cell2mat转换为矩阵
5、最后,我们就可以在MATLAB中对数组进行处理啦,最后传出的数组经过相似的操作即可在Python中继续进行运算。