triton/examples/indexing.cpp

#include "isaac/array.h"

namespace sc = isaac;

int main()
{
    static const char * sline = "--------------------";
    static const char * dline = "====================";

    std::cout << dline << std::endl;
    std::cout << "Tutorial: Indexing " << std::endl;
    std::cout << dline << std::endl;

    sc::int_t M = 5, N = 12;

    std::vector<float> data(M*N);
    for(unsigned int i = 0 ; i < data.size(); ++i)
      data[i] = i;
    sc::array A = sc::array(M, N, data);

    std::cout << "A:" << std::endl;
    std::cout << sline << std::endl;
    std::cout << A << std::endl;

//    std::cout << sline << std::endl;
//    std::cout << "A[3, 2:end]:" << A(3, {2,sc::end}) << std::endl;

//    std::cout << sline << std::endl;
//    std::cout << "A[2:end, 4]:" << A({2,sc::end}, 4) << std::endl;

//    std::cout << sline << std::endl;
//    std::cout << "diag(A,  1): " << sc::diag(A, 1) << std::endl;

//    std::cout << sline << std::endl;
//    std::cout << "diag(A, -7): " << sc::diag(A, -7) << std::endl;
}