如何在矩阵中添加尾随单例维度

时间:2015-09-15 10:08:34

标签: matlab mex

正如标题所说,如何在Matlab中向矩阵添加尾随单例维度?

A=ones(3,3,1);给出3x3矩阵,而A=ones(1,3,3);给出1x3x3矩阵

添加到特定问题:

我有一个应用程序,我有N个MxM矩阵,我需要堆叠它们,结果成为MxMxN矩阵。但是,N可以是1,如果是,我需要矩阵为MxMx1。

注意:我知道这对Matlab脚本没什么意义,因为Loren suggested,在Matlab中有无限的"单例"非单身之后的尺寸。但是,在mex环境中,这是使用mxGetNumberOfDimensions的情况。

2 个答案:

答案 0 :(得分:7)

即使从MEX API的角度来看,MATLAB也会删除尾随单例。 docs for mxSetDimensions同样多说:

  

MATLAB®自动删除dims参数中指定的任何尾随单例维度。例如,如果ndim等于5且dims等于[4 1 7 1 1],则生成的数组的尺寸为4 x 1乘7。

再次,这来自mxSetDimensions文档。

以下是使用mxSetDimensionsmxCreateNumericArray

尝试此操作的测试
// mexSizeTest.cpp
#include "mex.h"
void mexFunction(int nlhs, mxArray *plhs[],int nrhs, const mxArray *prhs[])
{
    if (nrhs!=1)
        mexErrMsgTxt("Just one input please.");

    mxArray *M = mxDuplicateArray(prhs[0]);
    const mwSize ndimsOrig = mxGetNumberOfDimensions(M);
    const mwSize *dimsOrig = mxGetDimensions(M);

    mexPrintf("Number of dimensions (input): %d\n\n", ndimsOrig);

    const mwSize ndims = ndimsOrig + 1;
    mwSize *dims = (mwSize*) mxMalloc(ndims*sizeof(mwSize));
    for (int i=0; i<ndimsOrig; ++i) dims[i] = dimsOrig[i];
    dims[ndims-1] = 1;

    mexPrintf("Target dimensions: [");
    for (int i=0; i<ndims-1; ++i) mexPrintf("%d ",dims[i]);
    mexPrintf("%d]\n\n",dims[ndims-1]);

    mexPrintf("Reshaping to #dims = %d with trailing singleton.\n", ndims);
    mxSetDimensions(M, dims, ndims);
    mexPrintf("Number of Dimensions: %d\n\n", mxGetNumberOfDimensions(M));

    // Let's be dangerous to see if the 1 is just hiding under the hood
    const mwSize *dimsSet = mxGetDimensions(M);
    mexPrintf("Being dangerous: %d\n\n", dimsSet[ndims-1]); // !!!
    mxDestroyArray(M);

    mexPrintf("Creating fresh mxArray of #dims = %d with trailing singleton.\n",
            ndims);
    M = mxCreateNumericArray(ndims, dims, mxDOUBLE_CLASS, mxREAL);
    mexPrintf("Number of Dimensions: %d\n",mxGetNumberOfDimensions(M));
    mxDestroyArray(M); mxFree(dims);
}

MATLAB测试:

>> M = rand(24,55,1);
>> size(M)
ans =
    24    55
>> ndims(M)
ans =
     2
>> size(M,454235) % note we can check the 454235th dimension
ans =
     1

附有size(M,454235)测试的旁注:这是ndim文档在说Trailing singleton dimensions are ignored时的含义。他们真的没有被忽视,他们只是不在那里!

MEX测试(mexSizeTest.cpp):

>> mexSizeTest(M)
Number of dimensions (input): 2

Target dimensions: [24 55 1]

Reshaping to #dims = 3 with trailing singleton.
Number of Dimensions: 2

Being dangerous: -994713024

Creating fresh mxArray of #dims = 3 with trailing singleton.
Number of Dimensions: 2

我建议您调整代码以处理mxGetNumberOfDimensions返回2的情况。

答案 1 :(得分:4)

您可以使用permute注入单身人士。

A = rand(50,50); %50 x 50
B = permute(A, [1 3 2]); %50 x 1 x 50

3指的是默认的尾随单例。 “4”不引用任何内容,因此您需要迭代多次进样(您也不能重复“3”)。