wei-ku-1.cpp

来自「数值计算工具库,C语言编写的,可以直接调用.」· C++ 代码 · 共 43 行

#include "testsuite.h"
#include <blitz/array.h>

BZ_USING_NAMESPACE(blitz)

int main()
{
  BZ_USING_NAMESPACE(blitz::tensor);
  Range all = Range::all();

  {
    Array<int,5> A(6,5,4,3,2);
    int ld_begin = 1;
    int ld_end = 3;
    Array<int,5> B(6,5,2,3,2);

    A = 11111 + 10000*i + 1000*j + 100*k + 10*l + m;

    B = A(all, all, Range(ld_begin, ld_end-1), all, all);
    // cout << B << endl;    

    BZTEST(B(0,0,0,0,0) = 11111);
    BZTEST(B(1,1,1,1,1) = 22322);
    BZTEST(B(5,4,1,2,1) = 65332);
  }

  {
    Array<int,5> B(10,1,25,6,6);
    Array<int,5> A(10,1,125,6,6);
    int ld_begin = 1, ld_end = 26;
    A = 11111 + 10000*i + 1000*j + 100*k + 10*l + m;
    B = A(all, all, Range(ld_begin, ld_end-1), all, all);

cout << B(0,0,0,0,0) << endl
     << B(1,0,1,1,1) << endl
     << B(5,0,3,4,5) << endl;

cout << B << endl;
  }
    return 0;
}