Files
triton/examples/indexing.cpp

46 lines
1.2 KiB
C++
Raw Normal View History

2015-10-08 20:42:37 -04:00
#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 = 10, N = 7;
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 << std::endl;
std::cout << "A[3, 2:end]:" << std::endl;
std::cout << sline << std::endl;
std::cout << A(3, {2,sc::end}) << std::endl;
std::cout << std::endl;
std::cout << "A[2:end, 4]:" << std::endl;
std::cout << sline << std::endl;
std::cout << A({2,sc::end}, 4) << std::endl;
std::cout << std::endl;
std::cout << "diag(A, 1): " << std::endl;
std::cout << sline << std::endl;
std::cout << sc::diag(A, 1) << std::endl;
std::cout << std::endl;
std::cout << "diag(A, -7): " << std::endl;
std::cout << sline << std::endl;
std::cout << sc::diag(A, -7) << std::endl;
std::cout << std::endl;
}