Как справиться с большим пакетным умножением в массиве огня - PullRequest
1 голос
/ 11 февраля 2020

Я новичок в arrayfire, и в настоящее время у меня возникли некоторые проблемы. В настоящее время я делаю большой пакет матричных умножений, как показано ниже, но мне не хватает памяти. Может ли кто-нибудь показать мне пример кода для решения этой проблемы?

    int n = 60;
    int m = 18000;
    int k = 600;
    int t = 200;            
    af::array A = randu(m, k, t); 
    af::array B = randu(k, n);   
    af::array C = af::constant(0, m, n, t); 
    C = matmul(A, B);
...