v0.13.2
Loading...
Searching...
No Matches
Tensor3_times_Dg.hpp
Go to the documentation of this file.
1/* Contracts a Tensor3 with a Dg, yielding a typename promote<T,U>::V,
2 Tensor2, or Tensor4. */
3
4#pragma once
5
6namespace FTensor
7{
8 /* A(i,j,k)*B(i,j,k) */
9
10 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
11 char j, char k, int Current_Dim0, int Current_Dim1,
12 int Current_Dim2>
13 typename promote<T, U>::V
14 T3_times_T3dg_012(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
15 const Dg_Expr<B, U, Dim, Dim2, i, j, k> &b,
16 const Number<Current_Dim0> &, const Number<Current_Dim1> &,
17 const Number<Current_Dim2> &)
18 {
19 return a(Current_Dim0 - 1, Current_Dim1 - 1, Current_Dim2 - 1)
20 * b(Current_Dim0 - 1, Current_Dim1 - 1, Current_Dim2 - 1)
21 + T3_times_T3dg_012(a, b, Number<Current_Dim0>(),
22 Number<Current_Dim1 - 1>(),
23 Number<Current_Dim2>());
24 }
25
26 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
27 char j, char k, int Current_Dim0, int Current_Dim2>
28 typename promote<T, U>::V
29 T3_times_T3dg_012(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
30 const Dg_Expr<B, U, Dim, Dim2, i, j, k> &b,
31 const Number<Current_Dim0> &, const Number<1> &,
32 const Number<Current_Dim2> &)
33 {
34 return a(Current_Dim0 - 1, 0, Current_Dim2 - 1)
35 * b(Current_Dim0 - 1, 0, Current_Dim2 - 1)
36 + T3_times_T3dg_012(a, b, Number<Current_Dim0>(), Number<Dim>(),
37 Number<Current_Dim2 - 1>());
38 }
39
40 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
41 char j, char k, int Current_Dim0>
42 typename promote<T, U>::V
43 T3_times_T3dg_012(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
44 const Dg_Expr<B, U, Dim, Dim2, i, j, k> &b,
45 const Number<Current_Dim0> &, const Number<1> &,
46 const Number<1> &)
47 {
48 return a(Current_Dim0 - 1, 0, 0) * b(Current_Dim0 - 1, 0, 0)
49 + T3_times_T3dg_012(a, b, Number<Current_Dim0 - 1>(), Number<Dim>(),
50 Number<Dim2>());
51 }
52
53 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
54 char j, char k>
55 typename promote<T, U>::V
56 T3_times_T3dg_012(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
57 const Dg_Expr<B, U, Dim, Dim2, i, j, k> &b,
58 const Number<1> &, const Number<1> &, const Number<1> &)
59 {
60 return a(0, 0, 0) * b(0, 0, 0);
61 }
62
63 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
64 char j, char k>
65 typename promote<T, U>::V
66 operator*(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
67 const Dg_Expr<B, U, Dim, Dim2, i, j, k> &b)
68 {
69 return T3_times_T3dg_012(a, b, Number<Dim>(), Number<Dim>(),
70 Number<Dim2>());
71 }
72
73 /* A(i,j,k)*B(k,i,j) */
74
75 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
76 char j, char k, int Current_Dim0, int Current_Dim1,
77 int Current_Dim2>
78 typename promote<T, U>::V
79 T3_times_T3dg_201(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
80 const Dg_Expr<B, U, Dim, Dim1, k, i, j> &b,
81 const Number<Current_Dim0> &, const Number<Current_Dim1> &,
82 const Number<Current_Dim2> &)
83 {
84 return a(Current_Dim0 - 1, Current_Dim1 - 1, Current_Dim2 - 1)
85 * b(Current_Dim2 - 1, Current_Dim0 - 1, Current_Dim1 - 1)
86 + T3_times_T3dg_201(a, b, Number<Current_Dim0>(),
87 Number<Current_Dim1 - 1>(),
88 Number<Current_Dim2>());
89 }
90
91 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
92 char j, char k, int Current_Dim0, int Current_Dim2>
93 typename promote<T, U>::V
94 T3_times_T3dg_201(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
95 const Dg_Expr<B, U, Dim, Dim1, k, i, j> &b,
96 const Number<Current_Dim0> &, const Number<1> &,
97 const Number<Current_Dim2> &)
98 {
99 return a(Current_Dim0 - 1, 0, Current_Dim2 - 1)
100 * b(Current_Dim2 - 1, Current_Dim0 - 1, 0)
101 + T3_times_T3dg_201(a, b, Number<Current_Dim0>(), Number<Dim1>(),
102 Number<Current_Dim2 - 1>());
103 }
104
105 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
106 char j, char k, int Current_Dim0>
107 typename promote<T, U>::V
108 T3_times_T3dg_201(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
109 const Dg_Expr<B, U, Dim, Dim1, k, i, j> &b,
110 const Number<Current_Dim0> &, const Number<1> &,
111 const Number<1> &)
112 {
113 return a(Current_Dim0 - 1, 0, 0) * b(0, Current_Dim0 - 1, 0)
114 + T3_times_T3dg_201(a, b, Number<Current_Dim0 - 1>(),
115 Number<Dim1>(), Number<Dim>());
116 }
117
118 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
119 char j, char k>
120 typename promote<T, U>::V
121 T3_times_T3dg_201(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
122 const Dg_Expr<B, U, Dim, Dim1, k, i, j> &b,
123 const Number<1> &, const Number<1> &, const Number<1> &)
124 {
125 return a(0, 0, 0) * b(0, 0, 0);
126 }
127
128 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
129 char j, char k>
130 typename promote<T, U>::V
131 operator*(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
132 const Dg_Expr<B, U, Dim, Dim1, k, i, j> &b)
133 {
134 return T3_times_T3dg_201(a, b, Number<Dim>(), Number<Dim1>(),
135 Number<Dim>());
136 }
137
138 /* A(i,j,k)*B(j,k,i) */
139
140 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
141 char j, char k, int Current_Dim0, int Current_Dim1,
142 int Current_Dim2>
143 typename promote<T, U>::V
144 T3_times_T3dg_120(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
145 const Dg_Expr<B, U, Dim, Dim0, j, k, i> &b,
146 const Number<Current_Dim0> &, const Number<Current_Dim1> &,
147 const Number<Current_Dim2> &)
148 {
149 return a(Current_Dim0 - 1, Current_Dim1 - 1, Current_Dim2 - 1)
150 * b(Current_Dim1 - 1, Current_Dim2 - 1, Current_Dim0 - 1)
151 + T3_times_T3dg_120(a, b, Number<Current_Dim0>(),
152 Number<Current_Dim1 - 1>(),
153 Number<Current_Dim2>());
154 }
155
156 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
157 char j, char k, int Current_Dim0, int Current_Dim2>
158 typename promote<T, U>::V
159 T3_times_T3dg_120(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
160 const Dg_Expr<B, U, Dim, Dim0, j, k, i> &b,
161 const Number<Current_Dim0> &, const Number<1> &,
162 const Number<Current_Dim2> &)
163 {
164 return a(Current_Dim0 - 1, 0, Current_Dim2 - 1)
165 * b(0, Current_Dim2 - 1, Current_Dim0 - 1)
166 + T3_times_T3dg_120(a, b, Number<Current_Dim0>(), Number<Dim>(),
167 Number<Current_Dim2 - 1>());
168 }
169
170 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
171 char j, char k, int Current_Dim0>
172 typename promote<T, U>::V
173 T3_times_T3dg_120(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
174 const Dg_Expr<B, U, Dim, Dim0, j, k, i> &b,
175 const Number<Current_Dim0> &, const Number<1> &,
176 const Number<1> &)
177 {
178 return a(Current_Dim0 - 1, 0, 0) * b(0, 0, Current_Dim0 - 1)
179 + T3_times_T3dg_120(a, b, Number<Current_Dim0 - 1>(), Number<Dim>(),
180 Number<Dim>());
181 }
182
183 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
184 char j, char k>
185 typename promote<T, U>::V
186 T3_times_T3dg_120(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
187 const Dg_Expr<B, U, Dim, Dim0, j, k, i> &b,
188 const Number<1> &, const Number<1> &, const Number<1> &)
189 {
190 return a(0, 0, 0) * b(0, 0, 0);
191 }
192
193 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
194 char j, char k>
195 typename promote<T, U>::V
196 operator*(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
197 const Dg_Expr<B, U, Dim, Dim0, j, k, i> &b)
198 {
199 return T3_times_T3dg_120(a, b, Number<Dim0>(), Number<Dim>(),
200 Number<Dim>());
201 }
202
203 /* A(i,j,k)*B(j,i,k) */
204
205 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
206 char j, char k, int Current_Dim0, int Current_Dim1,
207 int Current_Dim2>
208 typename promote<T, U>::V
209 T3_times_T3dg_102(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
210 const Dg_Expr<B, U, Dim, Dim2, j, i, k> &b,
211 const Number<Current_Dim0> &, const Number<Current_Dim1> &,
212 const Number<Current_Dim2> &)
213 {
214 return a(Current_Dim0 - 1, Current_Dim1 - 1, Current_Dim2 - 1)
215 * b(Current_Dim1 - 1, Current_Dim0 - 1, Current_Dim2 - 1)
216 + T3_times_T3dg_102(a, b, Number<Current_Dim0>(),
217 Number<Current_Dim1 - 1>(),
218 Number<Current_Dim2>());
219 }
220
221 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
222 char j, char k, int Current_Dim0, int Current_Dim2>
223 typename promote<T, U>::V
224 T3_times_T3dg_102(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
225 const Dg_Expr<B, U, Dim, Dim2, j, i, k> &b,
226 const Number<Current_Dim0> &, const Number<1> &,
227 const Number<Current_Dim2> &)
228 {
229 return a(Current_Dim0 - 1, 0, Current_Dim2 - 1)
230 * b(0, Current_Dim0 - 1, Current_Dim2 - 1)
231 + T3_times_T3dg_102(a, b, Number<Current_Dim0>(), Number<Dim>(),
232 Number<Current_Dim2 - 1>());
233 }
234
235 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
236 char j, char k, int Current_Dim0>
237 typename promote<T, U>::V
238 T3_times_T3dg_102(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
239 const Dg_Expr<B, U, Dim, Dim2, j, i, k> &b,
240 const Number<Current_Dim0> &, const Number<1> &,
241 const Number<1> &)
242 {
243 return a(Current_Dim0 - 1, 0, 0) * b(0, Current_Dim0 - 1, 0)
244 + T3_times_T3dg_102(a, b, Number<Current_Dim0 - 1>(), Number<Dim>(),
245 Number<Dim2>());
246 }
247
248 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
249 char j, char k>
250 typename promote<T, U>::V
251 T3_times_T3dg_102(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
252 const Dg_Expr<B, U, Dim, Dim2, j, i, k> &b,
253 const Number<1> &, const Number<1> &, const Number<1> &)
254 {
255 return a(0, 0, 0) * b(0, 0, 0);
256 }
257
258 template <class A, class B, class T, class U, int Dim, int Dim2, char i,
259 char j, char k>
260 typename promote<T, U>::V
261 operator*(const Tensor3_Expr<A, T, Dim, Dim, Dim2, i, j, k> &a,
262 const Dg_Expr<B, U, Dim, Dim2, j, i, k> &b)
263 {
264 return T3_times_T3dg_102(a, b, Number<Dim>(), Number<Dim>(),
265 Number<Dim2>());
266 }
267
268 /* A(i,j,k)*B(k,j,i) */
269
270 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
271 char j, char k, int Current_Dim0, int Current_Dim1,
272 int Current_Dim2>
273 typename promote<T, U>::V
274 T3_times_T3dg_210(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
275 const Dg_Expr<B, U, Dim, Dim0, k, j, i> &b,
276 const Number<Current_Dim0> &, const Number<Current_Dim1> &,
277 const Number<Current_Dim2> &)
278 {
279 return a(Current_Dim0 - 1, Current_Dim1 - 1, Current_Dim2 - 1)
280 * b(Current_Dim2 - 1, Current_Dim1 - 1, Current_Dim0 - 1)
281 + T3_times_T3dg_210(a, b, Number<Current_Dim0>(),
282 Number<Current_Dim1 - 1>(),
283 Number<Current_Dim2>());
284 }
285
286 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
287 char j, char k, int Current_Dim0, int Current_Dim2>
288 typename promote<T, U>::V
289 T3_times_T3dg_210(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
290 const Dg_Expr<B, U, Dim, Dim0, k, j, i> &b,
291 const Number<Current_Dim0> &, const Number<1> &,
292 const Number<Current_Dim2> &)
293 {
294 return a(Current_Dim0 - 1, 0, Current_Dim2 - 1)
295 * b(Current_Dim2 - 1, 0, Current_Dim0 - 1)
296 + T3_times_T3dg_210(a, b, Number<Current_Dim0>(), Number<Dim>(),
297 Number<Current_Dim2 - 1>());
298 }
299
300 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
301 char j, char k, int Current_Dim0>
302 typename promote<T, U>::V
303 T3_times_T3dg_210(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
304 const Dg_Expr<B, U, Dim, Dim0, k, j, i> &b,
305 const Number<Current_Dim0> &, const Number<1> &,
306 const Number<1> &)
307 {
308 return a(Current_Dim0 - 1, 0, 0) * b(0, 0, Current_Dim0 - 1)
309 + T3_times_T3dg_210(a, b, Number<Current_Dim0 - 1>(), Number<Dim>(),
310 Number<Dim>());
311 }
312
313 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
314 char j, char k>
315 typename promote<T, U>::V
316 T3_times_T3dg_210(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
317 const Dg_Expr<B, U, Dim, Dim0, k, j, i> &b,
318 const Number<1> &, const Number<1> &, const Number<1> &)
319 {
320 return a(0, 0, 0) * b(0, 0, 0);
321 }
322
323 template <class A, class B, class T, class U, int Dim0, int Dim, char i,
324 char j, char k>
325 typename promote<T, U>::V
326 operator*(const Tensor3_Expr<A, T, Dim0, Dim, Dim, i, j, k> &a,
327 const Dg_Expr<B, U, Dim, Dim0, k, j, i> &b)
328 {
329 return T3_times_T3dg_210(a, b, Number<Dim0>(), Number<Dim>(),
330 Number<Dim>());
331 }
332
333 /* A(i,j,k)*B(i,k,j) */
334
335 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
336 char j, char k, int Current_Dim0, int Current_Dim1,
337 int Current_Dim2>
338 typename promote<T, U>::V
339 T3_times_T3dg_021(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
340 const Dg_Expr<B, U, Dim, Dim1, i, k, j> &b,
341 const Number<Current_Dim0> &, const Number<Current_Dim1> &,
342 const Number<Current_Dim2> &)
343 {
344 return a(Current_Dim0 - 1, Current_Dim1 - 1, Current_Dim2 - 1)
345 * b(Current_Dim0 - 1, Current_Dim2 - 1, Current_Dim1 - 1)
346 + T3_times_T3dg_021(a, b, Number<Current_Dim0>(),
347 Number<Current_Dim1 - 1>(),
348 Number<Current_Dim2>());
349 }
350
351 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
352 char j, char k, int Current_Dim0, int Current_Dim2>
353 typename promote<T, U>::V
354 T3_times_T3dg_021(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
355 const Dg_Expr<B, U, Dim, Dim1, i, k, j> &b,
356 const Number<Current_Dim0> &, const Number<1> &,
357 const Number<Current_Dim2> &)
358 {
359 return a(Current_Dim0 - 1, 0, Current_Dim2 - 1)
360 * b(Current_Dim0 - 1, Current_Dim2 - 1, 0)
361 + T3_times_T3dg_021(a, b, Number<Current_Dim0>(), Number<Dim1>(),
362 Number<Current_Dim2 - 1>());
363 }
364
365 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
366 char j, char k, int Current_Dim0>
367 typename promote<T, U>::V
368 T3_times_T3dg_021(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
369 const Dg_Expr<B, U, Dim, Dim1, i, k, j> &b,
370 const Number<Current_Dim0> &, const Number<1> &,
371 const Number<1> &)
372 {
373 return a(Current_Dim0 - 1, 0, 0) * b(Current_Dim0 - 1, 0, 0)
374 + T3_times_T3dg_021(a, b, Number<Current_Dim0 - 1>(),
375 Number<Dim1>(), Number<Dim>());
376 }
377
378 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
379 char j, char k>
380 typename promote<T, U>::V
381 T3_times_T3dg_021(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
382 const Dg_Expr<B, U, Dim, Dim1, i, k, j> &b,
383 const Number<1> &, const Number<1> &, const Number<1> &)
384 {
385 return a(0, 0, 0) * b(0, 0, 0);
386 }
387
388 template <class A, class B, class T, class U, int Dim, int Dim1, char i,
389 char j, char k>
390 typename promote<T, U>::V
391 operator*(const Tensor3_Expr<A, T, Dim, Dim1, Dim, i, j, k> &a,
392 const Dg_Expr<B, U, Dim, Dim1, i, k, j> &b)
393 {
394 return T3_times_T3dg_021(a, b, Number<Dim>(), Number<Dim1>(),
395 Number<Dim>());
396 }
397
398 /* A(j,i,k)*B(k,l,j) */
399
400 template <class A, class B, class T, class U, int Dim2, int Dim1, int Dim01,
401 char i, char j, char k, char l>
402 class Tensor3_times_Dg_02_20
403 {
404 Tensor3_Expr<A, T, Dim2, Dim1, Dim01, j, i, k> iterA;
405 Dg_Expr<B, U, Dim01, Dim2, k, l, j> iterB;
406
407 template <int Current_Dim0, int Current_Dim1>
408 typename promote<T, U>::V
409 eval(const int N1, const int N2, const Number<Current_Dim0> &,
410 const Number<Current_Dim1> &) const
411 {
412 return iterA(Current_Dim0 - 1, N1, Current_Dim1 - 1)
413 * iterB(Current_Dim1 - 1, N2, Current_Dim0 - 1)
414 + eval(N1, N2, Number<Current_Dim0 - 1>(),
415 Number<Current_Dim1>());
416 }
417 template <int Current_Dim1>
418 typename promote<T, U>::V
419 eval(const int N1, const int N2, const Number<1> &,
420 const Number<Current_Dim1> &) const
421 {
422 return iterA(0, N1, Current_Dim1 - 1) * iterB(Current_Dim1 - 1, N2, 0)
423 + eval(N1, N2, Number<Dim2>(), Number<Current_Dim1 - 1>());
424 }
425 typename promote<T, U>::V eval(const int N1, const int N2,
426 const Number<1> &, const Number<1> &) const
427 {
428 return iterA(0, N1, 0) * iterB(0, N2, 0);
429 }
430
431 public:
432 Tensor3_times_Dg_02_20(
433 const Tensor3_Expr<A, T, Dim2, Dim1, Dim01, j, i, k> &a,
434 const Dg_Expr<B, U, Dim01, Dim2, k, l, j> &b)
435 : iterA(a), iterB(b)
436 {}
437 typename promote<T, U>::V operator()(const int N1, const int N2) const
438 {
439 return eval(N1, N2, Number<Dim2>(), Number<Dim01>());
440 }
441 };
442
443 template <class A, class B, class T, class U, int Dim2, int Dim1, int Dim01,
444 char i, char j, char k, char l>
445 Tensor2_Expr<Tensor3_times_Dg_02_20<A, B, T, U, Dim2, Dim1, Dim01, i, j, k, l>,
446 typename promote<T, U>::V, Dim1, Dim01, i, l>
447 operator*(const Tensor3_Expr<A, T, Dim2, Dim1, Dim01, j, i, k> &a,
448 const Dg_Expr<B, U, Dim01, Dim2, k, l, j> &b)
449 {
450 using TensorExpr
451 = Tensor3_times_Dg_02_20<A, B, T, U, Dim2, Dim1, Dim01, i, j, k, l>;
452 return Tensor2_Expr<TensorExpr, typename promote<T, U>::V, Dim1, Dim01, i,
453 l>(TensorExpr(a, b));
454 }
455
456 /* B(k,l,j)*A(j,i,k) */
457
458 /* A(i,j,k)*B(l,k,m) -> Tensor4 */
459
460 template <class A, class B, class T, class U, int Dim0, int Dim1, int Dim2,
461 int Dim, char i, char j, char k, char l, char m>
462 class Tensor3_times_Dg
463 {
464 Tensor3_Expr<A, T, Dim0, Dim1, Dim, i, j, k> iterA;
465 Dg_Expr<B, U, Dim, Dim2, l, k, m> iterB;
466
467 template <int Current_Dim>
468 typename promote<T, U>::V
469 eval(const int N1, const int N2, const int N3, const int N4,
470 const Number<Current_Dim> &) const
471 {
472 return iterA(N1, N2, Current_Dim - 1) * iterB(N3, Current_Dim - 1, N4)
473 + eval(N1, N2, N3, N4, Number<Current_Dim - 1>());
474 }
475 typename promote<T, U>::V eval(const int N1, const int N2, const int N3,
476 const int N4, const Number<1> &) const
477 {
478 return iterA(N1, N2, 0) * iterB(N3, 0, N4);
479 }
480
481 public:
482 typename promote<T, U>::V
483 operator()(const int N1, const int N2, const int N3, const int N4) const
484 {
485 return eval(N1, N2, N3, N4, Number<Dim>());
486 }
487 Tensor3_times_Dg(const Tensor3_Expr<A, T, Dim0, Dim1, Dim, i, j, k> &a,
488 const Dg_Expr<B, U, Dim, Dim2, l, k, m> &b)
489 : iterA(a), iterB(b)
490 {}
491 };
492
493 template <class A, class B, class T, class U, int Dim0, int Dim1, int Dim2,
494 int Dim, char i, char j, char k, char l, char m>
495 Tensor4_Expr<
496 Tensor3_times_Dg<A, B, T, U, Dim0, Dim1, Dim, Dim2, i, j, k, l, m>,
497 typename promote<T, U>::V, Dim0, Dim1, Dim, Dim2, i, j, l, m>
498 operator*(const Tensor3_Expr<A, T, Dim0, Dim1, Dim, i, j, k> &a,
499 const Dg_Expr<B, U, Dim, Dim2, l, k, m> &b)
500 {
501 using TensorExpr
502 = Tensor3_times_Dg<A, B, T, U, Dim0, Dim1, Dim2, Dim, i, j, k, l, m>;
503 return Tensor4_Expr<TensorExpr, typename promote<T, U>::V, Dim0, Dim1, Dim,
504 Dim2, i, j, l, m>(TensorExpr(a, b));
505 }
506
507 /* Yield Tensor 2*/
508
509 /* A(i,j,k) * B(i,j,l) -> Tensor2 */
510
511 template <class A, class B, class T, class U, int Dim03, int Dim14, int Dim2,
512 int Dim5, char i, char j, char k, char l>
513 class Tensor3_times_Dg_12_12 {
514 Tensor3_Expr<A, T, Dim03, Dim14, Dim2, i, j, k> iterA;
515 Dg_Expr<B, U, Dim03, Dim5, i, j, l> iterB;
516
517 template <int CurrentDim0, int CurrentDim1>
518 typename promote<T, U>::V eval(const int N1, const int N2,
519 const Number<CurrentDim0> &,
520 const Number<CurrentDim1> &) const {
521 return iterA(CurrentDim0 - 1, CurrentDim1 - 1, N1) *
522 iterB(CurrentDim0 - 1, CurrentDim1 - 1, N2) +
523 eval(N1, N2, Number<CurrentDim0>(), Number<CurrentDim1 - 1>());
524 }
525 template <int CurrentDim0>
526 typename promote<T, U>::V eval(const int N1, const int N2,
527 const Number<CurrentDim0> &,
528 const Number<1> &) const {
529 return iterA(CurrentDim0 - 1, 0, N1) * iterB(CurrentDim0 - 1, 0, N2) +
530 eval(N1, N2, Number<CurrentDim0 - 1>(), Number<Dim14>());
531 }
532 typename promote<T, U>::V eval(const int N1, const int N2,
533 const Number<1> &, const Number<1> &) const {
534 return iterA(0, 0, N1) * iterB(0, 0, N2);
535 }
536
537 public:
538 Tensor3_times_Dg_12_12(
539 const Tensor3_Expr<A, T, Dim03, Dim14, Dim2, i, j, k> &a,
540 const Dg_Expr<B, U, Dim03, Dim5, i, j, l> &b)
541 : iterA(a), iterB(b) {
542
543 static_assert(Dim03 == Dim14, "Dimension 03 should be as Dim14");
544 }
545 typename promote<T, U>::V operator()(const int &N1, const int &N2) const {
546 return eval(N1, N2, Number<Dim03>(), Number<Dim14>());
547 }
548 };
549
550 template <class A, class B, class T, class U, int Dim03, int Dim14, int Dim2,
551 int Dim5, char i, char j, char k, char l>
552 Tensor2_Expr<
553 Tensor3_times_Dg_12_12<A, B, T, U, Dim03, Dim14, Dim2, Dim5, i, j, k, l>,
554 typename promote<T, U>::V, Dim2, Dim5, k, l>
555 operator*(const Tensor3_Expr<A, T, Dim03, Dim14, Dim2, i, j, k> &a,
556 const Dg_Expr<B, U, Dim03, Dim5, i, j, l> &b) {
557 using TensorExpr = Tensor3_times_Dg_12_12<A, B, T, U, Dim03, Dim14, Dim2,
558 Dim5, i, j, k, l>;
559 return Tensor2_Expr<TensorExpr, typename promote<T, U>::V, Dim2, Dim5, k,
560 l>(TensorExpr(a, b));
561 };
562
563 /* B(i,j,l)* A(i,j,k) -> Tensor2 */
564
565 template <class A, class B, class T, class U, int Dim03, int Dim14, int Dim2,
566 int Dim5, char i, char j, char k, char l>
567 Tensor2_Expr<
568 Tensor3_times_Dg_12_12<A, B, T, U, Dim03, Dim14, Dim2, Dim5, i, j, k, l>,
569 typename promote<T, U>::V, Dim2, Dim5, k, l>
570 operator*(const Dg_Expr<B, U, Dim03, Dim5, i, j, l> &b,
571 const Tensor3_Expr<A, T, Dim03, Dim14, Dim2, i, j, k> &a) {
572 using TensorExpr = Tensor3_times_Dg_12_12<A, B, T, U, Dim03, Dim14, Dim2,
573 Dim5, i, j, k, l>;
574 return Tensor2_Expr<TensorExpr, typename promote<T, U>::V, Dim2, Dim5, k,
575 l>(TensorExpr(a, b));
576 };
577}
N2
static Number< 2 > N2
Definition: BasicFeTools.hpp:101
N1
static Number< 1 > N1
Definition: BasicFeTools.hpp:100
a
constexpr double a
Definition: approx_sphere.cpp:30
FTensor::Dg_Expr
Definition: Dg_Expr.hpp:26
FTensor::Tensor2_Expr
Definition: Tensor2_Expr.hpp:27
FTensor::Tensor3_Expr
Definition: Tensor3_Expr.hpp:25
FTensor::Tensor3_times_Dg_02_20
Definition: Tensor3_times_Dg.hpp:403
FTensor::Tensor3_times_Dg_02_20::eval
promote< T, U >::V eval(const int N1, const int N2, const Number< 1 > &, const Number< 1 > &) const
Definition: Tensor3_times_Dg.hpp:425
FTensor::Tensor3_times_Dg_02_20::iterB
Dg_Expr< B, U, Dim01, Dim2, k, l, j > iterB
Definition: Tensor3_times_Dg.hpp:405
FTensor::Tensor3_times_Dg_02_20::eval
promote< T, U >::V eval(const int N1, const int N2, const Number< Current_Dim0 > &, const Number< Current_Dim1 > &) const
Definition: Tensor3_times_Dg.hpp:409
FTensor::Tensor3_times_Dg_02_20::eval
promote< T, U >::V eval(const int N1, const int N2, const Number< 1 > &, const Number< Current_Dim1 > &) const
Definition: Tensor3_times_Dg.hpp:419
FTensor::Tensor3_times_Dg_02_20::operator()
promote< T, U >::V operator()(const int N1, const int N2) const
Definition: Tensor3_times_Dg.hpp:437
FTensor::Tensor3_times_Dg_02_20::iterA
Tensor3_Expr< A, T, Dim2, Dim1, Dim01, j, i, k > iterA
Definition: Tensor3_times_Dg.hpp:404
FTensor::Tensor3_times_Dg_02_20::Tensor3_times_Dg_02_20
Tensor3_times_Dg_02_20(const Tensor3_Expr< A, T, Dim2, Dim1, Dim01, j, i, k > &a, const Dg_Expr< B, U, Dim01, Dim2, k, l, j > &b)
Definition: Tensor3_times_Dg.hpp:432
FTensor::Tensor3_times_Dg_12_12
Definition: Tensor3_times_Dg.hpp:513
FTensor::Tensor3_times_Dg_12_12::operator()
promote< T, U >::V operator()(const int &N1, const int &N2) const
Definition: Tensor3_times_Dg.hpp:545
FTensor::Tensor3_times_Dg_12_12::eval
promote< T, U >::V eval(const int N1, const int N2, const Number< CurrentDim0 > &, const Number< CurrentDim1 > &) const
Definition: Tensor3_times_Dg.hpp:518
FTensor::Tensor3_times_Dg_12_12::eval
promote< T, U >::V eval(const int N1, const int N2, const Number< CurrentDim0 > &, const Number< 1 > &) const
Definition: Tensor3_times_Dg.hpp:526
FTensor::Tensor3_times_Dg_12_12::iterA
Tensor3_Expr< A, T, Dim03, Dim14, Dim2, i, j, k > iterA
Definition: Tensor3_times_Dg.hpp:514
FTensor::Tensor3_times_Dg_12_12::Tensor3_times_Dg_12_12
Tensor3_times_Dg_12_12(const Tensor3_Expr< A, T, Dim03, Dim14, Dim2, i, j, k > &a, const Dg_Expr< B, U, Dim03, Dim5, i, j, l > &b)
Definition: Tensor3_times_Dg.hpp:538
FTensor::Tensor3_times_Dg_12_12::eval
promote< T, U >::V eval(const int N1, const int N2, const Number< 1 > &, const Number< 1 > &) const
Definition: Tensor3_times_Dg.hpp:532
FTensor::Tensor3_times_Dg_12_12::iterB
Dg_Expr< B, U, Dim03, Dim5, i, j, l > iterB
Definition: Tensor3_times_Dg.hpp:515
FTensor::Tensor3_times_Dg
Definition: Tensor3_times_Dg.hpp:463
FTensor::Tensor3_times_Dg::iterA
Tensor3_Expr< A, T, Dim0, Dim1, Dim, i, j, k > iterA
Definition: Tensor3_times_Dg.hpp:464
FTensor::Tensor3_times_Dg::Tensor3_times_Dg
Tensor3_times_Dg(const Tensor3_Expr< A, T, Dim0, Dim1, Dim, i, j, k > &a, const Dg_Expr< B, U, Dim, Dim2, l, k, m > &b)
Definition: Tensor3_times_Dg.hpp:487
FTensor::Tensor3_times_Dg::iterB
Dg_Expr< B, U, Dim, Dim2, l, k, m > iterB
Definition: Tensor3_times_Dg.hpp:465
FTensor::Tensor3_times_Dg::eval
promote< T, U >::V eval(const int N1, const int N2, const int N3, const int N4, const Number< Current_Dim > &) const
Definition: Tensor3_times_Dg.hpp:469
FTensor::Tensor3_times_Dg::eval
promote< T, U >::V eval(const int N1, const int N2, const int N3, const int N4, const Number< 1 > &) const
Definition: Tensor3_times_Dg.hpp:475
FTensor::Tensor3_times_Dg::operator()
promote< T, U >::V operator()(const int N1, const int N2, const int N3, const int N4) const
Definition: Tensor3_times_Dg.hpp:483
FTensor::Tensor4_Expr
Definition: Tensor4_Expr.hpp:26
m
FTensor::Index< 'm', SPACE_DIM > m
Definition: fluid_structure_eigenproblem.cpp:65
i
FTensor::Index< 'i', SPACE_DIM > i
Definition: hcurl_divergence_operator_2d.cpp:27
l
FTensor::Index< 'l', 3 > l
Definition: matrix_function.cpp:21
j
FTensor::Index< 'j', 3 > j
Definition: matrix_function.cpp:19
k
FTensor::Index< 'k', 3 > k
Definition: matrix_function.cpp:20
T
const double T
Definition: mixed_reac_diff.cpp:158
FTensor
Tensors class implemented by Walter Landry.
Definition: FTensor.hpp:51
FTensor::T3_times_T3dg_012
promote< T, U >::V T3_times_T3dg_012(const Tensor3_Expr< A, T, Dim, Dim, Dim2, i, j, k > &a, const Dg_Expr< B, U, Dim, Dim2, i, j, k > &b, const Number< Current_Dim0 > &, const Number< Current_Dim1 > &, const Number< Current_Dim2 > &)
Definition: Tensor3_times_Dg.hpp:14
FTensor::T3_times_T3dg_201
promote< T, U >::V T3_times_T3dg_201(const Tensor3_Expr< A, T, Dim, Dim1, Dim, i, j, k > &a, const Dg_Expr< B, U, Dim, Dim1, k, i, j > &b, const Number< Current_Dim0 > &, const Number< Current_Dim1 > &, const Number< Current_Dim2 > &)
Definition: Tensor3_times_Dg.hpp:79
FTensor::T3_times_T3dg_210
promote< T, U >::V T3_times_T3dg_210(const Tensor3_Expr< A, T, Dim0, Dim, Dim, i, j, k > &a, const Dg_Expr< B, U, Dim, Dim0, k, j, i > &b, const Number< Current_Dim0 > &, const Number< Current_Dim1 > &, const Number< Current_Dim2 > &)
Definition: Tensor3_times_Dg.hpp:274
FTensor::T3_times_T3dg_021
promote< T, U >::V T3_times_T3dg_021(const Tensor3_Expr< A, T, Dim, Dim1, Dim, i, j, k > &a, const Dg_Expr< B, U, Dim, Dim1, i, k, j > &b, const Number< Current_Dim0 > &, const Number< Current_Dim1 > &, const Number< Current_Dim2 > &)
Definition: Tensor3_times_Dg.hpp:339
FTensor::T3_times_T3dg_120
promote< T, U >::V T3_times_T3dg_120(const Tensor3_Expr< A, T, Dim0, Dim, Dim, i, j, k > &a, const Dg_Expr< B, U, Dim, Dim0, j, k, i > &b, const Number< Current_Dim0 > &, const Number< Current_Dim1 > &, const Number< Current_Dim2 > &)
Definition: Tensor3_times_Dg.hpp:144
FTensor::operator*
promote< T, U >::V operator*(const Ddg_Expr< A, T, Dim, Dim, i, j, k, l > &a, const Ddg_Expr< B, U, Dim, Dim, i, k, j, l > &b)
Definition: Ddg_times_Ddg.hpp:79
FTensor::T3_times_T3dg_102
promote< T, U >::V T3_times_T3dg_102(const Tensor3_Expr< A, T, Dim, Dim, Dim2, i, j, k > &a, const Dg_Expr< B, U, Dim, Dim2, j, i, k > &b, const Number< Current_Dim0 > &, const Number< Current_Dim1 > &, const Number< Current_Dim2 > &)
Definition: Tensor3_times_Dg.hpp:209
A
constexpr AssemblyType A
Definition: operators_tests.cpp:30
Generated by Doxygen 1.9.5 and hosted at University of Glasgow