Automatically set number of outputs of the gradient function

2 ビュー (過去 30 日間)

tiwwexx 2023 年 8 月 24 日

0
リンク

この質問への直接リンク

https://jp.mathworks.com/matlabcentral/answers/2012332-automatically-set-number-of-outputs-of-the-gradient-function

コメント済み: tiwwexx 2023 年 8 月 29 日

I have a loss function I define to minimize some gradient magnitude. However, the dimension of my input data will change so I want to make my gradient have a generalized output. I have something close to a solution but Dont know how to get rid of the hard coded 2nd line below.

grad_vec = zeros([size(DATA),numel(size(DATA))]);
[grad_vec(:,:,:,1),grad_vec(:,:,:,2),grad_vec(:,:,:,3)] = gradient(DATA);
grad_mag = (sum(abs(grad_vec).^n,numel(size(DATA)))).^(1/n);

3 件のコメント
1 件の古いコメントを表示1 件の古いコメントを非表示

Bruno Luong 2023 年 8 月 24 日

MATLAB Online で開く

it probably miss a second colon on lhs

if numel(size(DATA))==2
    [grad_vec(:,:,1),grad_vec(:,:,2)] = gradient(DATA);
...

Stephen23 2023 年 8 月 24 日

The solution is explained here:

https://www.mathworks.com/help/matlab/matlab_prog/comma-separated-lists.html

https://www.mathworks.com/matlabcentral/answers/1656435-tutorial-comma-separated-lists-and-how-to-use-them

サインインしてコメントする。

サインインしてこの質問に回答する。

採用された回答

Bruno Luong 2023 年 8 月 24 日

1
リンク

この回答への直接リンク

https://jp.mathworks.com/matlabcentral/answers/2012332-automatically-set-number-of-outputs-of-the-gradient-function#answer_1293122

編集済み: Bruno Luong 2023 年 8 月 24 日

MATLAB Online で開く

A = rand(10,10,10);
% This works for any dimensions of A
if isvector(A) % EDIT
    N = 1;
else
    N = ndims(A); % 3 in this example
end
G = cell(1,N);
[G{:}] = gradient(A);
G = cat(N+1,G{:});
size(G)
ans = 1×4
    10    10    10     3
p = 2; % whatever power in >=1
Gmag = sum(abs(G).^p,N+1).^(1/p);
size(Gmag)
ans = 1×3
    10    10    10