用matlab自定义一个求和函数

在MATLAB中，你可以很容易地创建一个自定义的求和函数。假设你想创建一个简单的求和函数，用于计算向量或矩阵的所有元素之和，可以按照以下步骤操作：

1. **定义函数**：首先，打开MATLAB并输入以下代码，这将定义一个名为`custom_sum`的函数，它接受一个输入变量（如`input_vector`）：

function result = custom_sum(input)
    % 检查输入是否为向量或矩阵
    if isscalar(input) || isvector(input)
        % 如果是向量，直接相加
        result = sum(input);
    elseif ismatrix(input)
        % 对于矩阵，对所有元素求和
        result = sum(sum(input));
    else
        error('Input must be a scalar, vector, or matrix.');
    end
end

2. **测试函数**：你可以现在就使用这个函数来测试它的效果，例如：

input_vector = [1, 2, 3, 4, 5];
sum_result = custom_sum(input_vector);
disp(['Custom sum of the input vector: ', num2str(sum_result)]);

% 或者对于矩阵
input_matrix = [1 2; 3 4];
matrix_result = custom_sum(input_matrix);
disp(['Custom sum of the input matrix: ', num2str(matrix_result)]);