Question

我有一个接收2个以上变量的函数

例如。

testFunction = @(x1, x2, x3, x4) x1.*x2.*x3.*x4;
testFunction2 = @(x1, x2, x3) sin(x1.*x2.^x3);

是否有像bsxfun这样的函数可以对具有2个以上输入的函数进行单例扩展？

bsxfun

的示例

binaryTestFunction = @(x1, x2) x1.*x2;
x1 = 9*eye(10);
x2 = 3*ones(10,1);
A = bsxfun(binaryTestFunction , x1 , x2);

扩展x2向量中的单例维度 bsxfun比repmat更快，也是高度迭代的，因此我想知道我的testFunction单例扩展是否可行。

Answer 1

您可以定义自己的类，以便自动使用bsxfun：

classdef bdouble < double
    %BDOUBLE double with automatic broadcasting enabled

    methods
        function obj=bdouble(data)
            obj = obj@double(data);
        end
        function r=plus(x,y)
            r=bsxfun(@plus,x,y);
        end
        function r=minus(x,y)
            r=bsxfun(@minus,x,y);
        end
        function r=power(x,y)
            r=bsxfun(@power,x,y);
        end
        function r=times(x,y)
            r=bsxfun(@times,x,y);
        end
        function r=rdivide(x,y)
            r=rdivide(@rdivide,x,y);
        end
        function r=ldivide(x,y)
            r=ldivide(@ldivide,x,y);
        end 
    end
end

用法：

 testFunction(bdouble([1:3]),bdouble([1:4]'),bdouble(cat(3,1,1)),bdouble(1))

如果你更喜欢bsxfun语法，你可以把它放在最上面：

function varargout=nbsxfun(fun,varargin)
varargout=cell(1:nargout);
for idx=1:numel(varargin)
    if isa(varargin{idx},'double')
        varargin{idx}=bdouble(varargin{idx});
    end
end
varargout{1:max(nargout,1)}=fun(varargin{:});
end

示例：

n=nbsxfun(testFunction,[1:3],[1:4]',cat(3,1,1),4)

Answer 2

嵌套功能：

testFunction = @(x1, x2, x3, x4) bsxfun(@times, bsxfun(@times, bsxfun(@times, x1, x2), x3), x4);
testFunction2 = @(x1, x2, x3) sin(bsxfun(@power, bsxfun(@times, x1, x2), x3);

请注意，您应该在函数内部实现bsxfun，而不是依赖于使用bsxfun调用函数的用户，因此无论如何，与函数的交互都是无缝的。

另请注意，当使用直接元素操作时，使用bsxfun之间的速度损失很小。

文件交换上有一个文件覆盖了数组运算符，因此bsxfun自动在所有运算符上使用，但我不记得它的名字。我使用了一段时间，但是你冒着用户不知道这已经实现的风险。

Matlab - 单输出扩展两个以上的输入

2 个答案: