Octave 中的指数矢量化

Question

我正在尝试获取一个通用数组 A，将其提升到向量 p 的每个成员的（元素方面）幂，然后对结果求和，最好是在向量运算中，这样结果与A 的大小相同。理想情况下，应该允许A 的任何大小/维度的数组。例如，如果

A = [0 1 ; 2 3]
p = [2 3]

我想要结果A.^p(1) + A.^p(2)，也就是[0 2 ; 12 36]，更优雅，A 的任何大小和p 的长度，并避免循环。

我想出了以下内容，它扩展到 A 的下一个更高维度，然后沿该维度求和：

sum(repmat(A,[ones(1,ndims(A)) length(p)]) .^ repmat(reshape(p,[ones(1,ndims(A)) length(p)]),size(A)),ndims(A)+1)

这在技术上似乎可行，但是....呃。有没有更清洁的方法来做到这一点？

Answer 1

假设p 是一个行向量：

result = reshape(sum(A(:) .^ p, 2), size(A));