Выберите подматрицу и векторизовать в одной команде в MATLAB

Question

Допустим, у нас есть массив A = ones(2, 2, 2) и другая матрица P = rand(4). Мне интересно, если это возможно, чтобы написать код

temp = A(:, :, 1); 
X = P * temp(:); 

в одну строку кода, чтобы сохранить память, потребляемую temp. Я попытался запустить

X = P * A(:, :, 1)(:); 

но это не работает. Я также пошарил с командой reshape, но не смог заставить ее работать.

Я не мог найти ответ, используя веб-сайт или этот форум. Можно ли делать то, что я ищу?

Спасибо за помощь,

Адриану

Answer 1

Вы могли бы сделать -

[m,n,r] = size(A); 
X = P*reshape(A(:,:,1),m*n,[]) 

Если вы делаете это итеративно по третьему измерению A, т.е. A(:, :, iter), где iter является итератором, вы можете получить все X в векторном формате в таком массиве -

X_all = P*reshape(A,m*n,[]) 

Answer 2

Рефайп должен работать. Попробуйте сделать это следующим образом:

X = P * reshape(A(:, :, 1), [], 1)