jiri/LA_library
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tensor: tested full contractions

Browse Source
This commit is contained in:
Jiri Pittner
2025-11-06 16:02:09 +01:00
parent b24cd24747
commit 7ef2d1b73d
2 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
t.cc
View File

@@ -3976,6 +3976,7 @@ cout <<t.dot(u)<<endl;
if(1)
{
//check full constractions
int r,n;
cin>>r>>n;
INDEXGROUP shape;
@@ -3986,13 +3987,17 @@ INDEXGROUP shape;
shape.range=n;
shape.offset=0;
}
Tensor<double> x(shape);
x.randomize(1.);
cout<<x;
Tensor<double> x(shape); x.randomize(1.);
Tensor<double> xf=x.flatten();
cout <<xf;
Tensor<double> y(shape); y.randomize(1.);
Tensor<double> yf=y.flatten();
cout <<x.dot(x) <<" "<< xf.dot(xf)<<endl;
Tensor<double> z = x.groupcontraction(0,y,0,1,false,true);
INDEXLIST cl(r);
for(int i=0; i<r; ++i) cl[i]={i,0};
Tensor<double> zf = xf.contractions(cl,yf,cl,1,false,true);
cout <<x.dot(y) <<" "<< xf.dot(yf)<< " "<<z<<" "<<zf<<endl;
}