请用Armadillo帮我实现matlab的gradient函数

可以使用Armadillo库来实现matlab中的gradient函数，以下是示例代码：

#include <iostream>
#include <armadillo>

using namespace std;
using namespace arma;

mat gradient(mat matrix) {
    int m = matrix.n_rows;
    int n = matrix.n_cols;
    mat grad(m, n);

    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            double dx, dy;
            if (i == 0) {
                dx = matrix(i + 1, j) - matrix(i, j);
            } else if (i == m - 1) {
                dx = matrix(i, j) - matrix(i - 1, j);
            } else {
                dx = (matrix(i + 1, j) - matrix(i - 1, j)) / 2;
            }

            if (j == 0) {
                dy = matrix(i, j + 1) - matrix(i, j);
            } else if (j == n - 1) {
                dy = matrix(i, j) - matrix(i, j - 1);
            } else {
                dy = (matrix(i, j + 1) - matrix(i, j - 1)) / 2;
            }

            grad(i, j) = sqrt(dx * dx + dy * dy);
        }
    }

    return grad;
}

int main() {
    mat matrix = { {1, 2, 3}, {4, 5, 6}, {7, 8, 9} };
    mat grad = gradient(matrix);

    cout << grad << endl;

    return 0;
}

这个函数使用了Armadillo库的mat类型来表示矩阵，并返回一个表示梯度的mat类型。对于每个元素，它计算了它周围元素的差异，并使用勾股定理计算出梯度大小。这个函数只能处理二维矩阵。