有没有办法沿着数组的不同维度自动获取渐变？

Question

我正在尝试在 Matlab 中找到一种方法来自动计算函数沿不同维度的梯度。

我找到了梯度函数here，但它需要我为每个维度写下单独的输出变量。 由于我在采样算法的中间计算梯度，我希望梯度计算能够随着更高维度的输入自动缩放。

基本上，如果F 是一个向量，我会想要： G = gradient(F) 成为具有单个条目的元胞数组：具有所有梯度值的向量。 但是，如果F 是二维数组，我希望G 是具有条目GX 和GY 的元胞数组，其中GX 是沿X 方向的梯度，GY 是沿X 方向的梯度Y 方向。

Answer 1

这样的事情可以解决问题：

nd = sum(size(F)>1);
G = cell(nd,1);
[G{:}] = gradient(F);

nd 是维数（简单地调用ndims 是不合适的，因为向量在 MATLAB 中有 2 维，所以一切都至少有 2 维）。 [G{:}] 构造将gradient 的一个输出参数分配给元胞数组的每个元素。