Реализовать шаблонную функцию нахождения максимального элемента главной диагонали квадратной матрицы

#include <iostream>
#include <ctime>
 
using std::cin;
using std::cout;
using std::endl;
 
template <typename T>
T max_element(T **matr, int n)
{
    T max = matr[0][0];
    for (int i=1; i<n; i++)   
        if (matr[i][i] > max)
            max = matr[i][i];
    return max;
}
 
int main()
{
    srand(time(NULL));    
    int n;
    cout << "Enter size: ";
    cin >> n;
    int **matrix = new int*[n];
    for (int i=0; i<n; i++)
    {
        matrix[i] = new int [n];
        for (int j=0; j<n; j++)
        {
            matrix[i][j] = rand() % 10;
            cout << matrix[i][j] << " ";
        }
        cout << endl;
    }
    cout << "Max element main diagonal of int matrix: " << max_element(matrix, n) << endl;
    
    for (int i=0; i<n; i++)
        delete[]matrix[i];
    delete[]matrix;
    
    double **matrixD = new double*[n];
    for (int i=0; i<n; i++)
    {
        matrixD[i] = new double[n];
        for (int j=0; j<n; j++)
        {
            matrixD[i][j] = (double)rand() / double(RAND_MAX) * 10.0;
            cout << matrixD[i][j] << " ";
        }
        cout << endl;
    }
    cout << "Max element main diagonal of double matrix: " << max_element(matrixD, n) << endl;
    for (int i=0; i<n; i++)
        delete[]matrixD[i];
    delete[]matrixD;
}