119 lines
1.6 KiB
C++
119 lines
1.6 KiB
C++
// Do the outer product of tensors.
|
|
|
|
#include "stdafx.h"
|
|
#include "defs.h"
|
|
|
|
static void outer_f(void);
|
|
|
|
void
|
|
eval_outer(void)
|
|
{
|
|
p1 = cdr(p1);
|
|
push(car(p1));
|
|
eval();
|
|
p1 = cdr(p1);
|
|
while (iscons(p1)) {
|
|
push(car(p1));
|
|
eval();
|
|
outer();
|
|
p1 = cdr(p1);
|
|
}
|
|
}
|
|
|
|
void
|
|
outer(void)
|
|
{
|
|
save();
|
|
p2 = pop();
|
|
p1 = pop();
|
|
if (istensor(p1) && istensor(p2))
|
|
outer_f();
|
|
else {
|
|
push(p1);
|
|
push(p2);
|
|
if (istensor(p1))
|
|
tensor_times_scalar();
|
|
else if (istensor(p2))
|
|
scalar_times_tensor();
|
|
else
|
|
multiply();
|
|
}
|
|
restore();
|
|
}
|
|
|
|
void
|
|
outer_f(void)
|
|
{
|
|
int i, j, k, ndim, nelem;
|
|
|
|
ndim = p1->u.tensor->ndim + p2->u.tensor->ndim;
|
|
|
|
if (ndim > MAXDIM)
|
|
stop("outer: rank of result exceeds maximum");
|
|
|
|
nelem = p1->u.tensor->nelem * p2->u.tensor->nelem;
|
|
|
|
p3 = alloc_tensor(nelem);
|
|
|
|
p3->u.tensor->ndim = ndim;
|
|
|
|
for (i = 0; i < p1->u.tensor->ndim; i++)
|
|
p3->u.tensor->dim[i] = p1->u.tensor->dim[i];
|
|
|
|
j = i;
|
|
|
|
for (i = 0; i < p2->u.tensor->ndim; i++)
|
|
p3->u.tensor->dim[j + i] = p2->u.tensor->dim[i];
|
|
|
|
k = 0;
|
|
|
|
for (i = 0; i < p1->u.tensor->nelem; i++)
|
|
for (j = 0; j < p2->u.tensor->nelem; j++) {
|
|
push(p1->u.tensor->elem[i]);
|
|
push(p2->u.tensor->elem[j]);
|
|
multiply();
|
|
p3->u.tensor->elem[k++] = pop();
|
|
}
|
|
|
|
push(p3);
|
|
}
|
|
|
|
static char *s[] = {
|
|
|
|
"outer(a,b)",
|
|
"a*b",
|
|
|
|
"outer(a,(b1,b2))",
|
|
"(a*b1,a*b2)",
|
|
|
|
"outer((a1,a2),b)",
|
|
"(a1*b,a2*b)",
|
|
|
|
"H33=hilbert(3)",
|
|
"",
|
|
|
|
"H44=hilbert(4)",
|
|
"",
|
|
|
|
"H55=hilbert(5)",
|
|
"",
|
|
|
|
"H3344=outer(H33,H44)",
|
|
"",
|
|
|
|
"H4455=outer(H44,H55)",
|
|
"",
|
|
|
|
"H33444455=outer(H33,H44,H44,H55)",
|
|
"",
|
|
|
|
"inner(H3344,H4455)-contract(H33444455,4,5)",
|
|
"0",
|
|
};
|
|
|
|
void
|
|
test_outer(void)
|
|
{
|
|
test(__FILE__, s, sizeof s / sizeof (char *));
|
|
}
|