如何在 GNU Octave 中为矩阵的每个元素调用具有求幂的自定义函数？

Question

试图找到一种方法来调用 GNU Octave 中矩阵中每个项目的自定义函数中使用的幂函数 ( ^ )。

我是一个初学者，我想这很简单，但我无法让它工作。

代码如下所示：

function result = the_function(the_val)
    result = (the_val - 5) ^ 2
endfunction

我试过这样称呼它：

>> A = [1,2,3];
>> the_function(A);
>> arrayfun(@the_function, A);
>> A .@the_function 2;

这些都不起作用（我认为最后一个根本不正确的语法），抛出错误：

error: for A^b, A must be a square matrix

我猜，这意味着它试图平方 矩阵，而不是其中的元素。

我应该怎么做？

非常感谢！

Answer 1

将函数调用为the_function(A) 是正确的，但您必须确保该函数可以处理向量输入。正如您所说，(the_val - 5)^2 尝试对矩阵求平方（因此，如果the_val 不是平方矩阵，则会出错）。要计算 element-wise 幂，请使用 use .^ 而不是 ^。

所以：在你的函数的定义中，你需要改变

result = (the_val-5)^2;

到

result = (the_val-5).^2;

作为附加说明，由于您的代码可以使用 标量 输入，因此您也可以使用 arrayfun 方法。正确的语法是（删除@）：

arrayfun(the_function, A)

然而，使用arrayfun 通常比定义你的函数要慢，这样它就可以直接处理向量输入（或“vectorizing”它）。所以，只要有可能，vectorize your function。这就是我上面的.^suggestion 所做的。