Esempio n. 1
0
def test_einsum_view1d(a):
    return einsum("i", a)
Esempio n. 2
0
def test_einsum_elemwise2d_T(a, b):
    return einsum("ij,ji->ij", a, b)
Esempio n. 3
0
def test_einsum_mm(a, b):
    return einsum("ij,jk->ik", a, b)
Esempio n. 4
0
def test_einsum_sum2d_01(a):
    return einsum("ij->", a)
Esempio n. 5
0
def test_einsum_elemwise2d(a, b):
    return einsum("ij,ij->ij", a, b)
Esempio n. 6
0
def test_einsum_sum1d(a):
    return einsum("i->", a)
Esempio n. 7
0
def test_einsum_sum2d_1(a):
    return einsum("ij->j", a)
Esempio n. 8
0
def test_einsum_diag(a):
    return einsum("ii->i", a)
Esempio n. 9
0
def test_einsum_trace(a):
    return einsum("ii", a)
Esempio n. 10
0
def test_einsum_view2d(a):
    return einsum("ij->ij", a)
Esempio n. 11
0
def test_einsum_transpose(a):
    return einsum("ij->ji", a)
Esempio n. 12
0
def test_einsum_outer(a, b):
    return einsum("i,k->ik", a, b)
Esempio n. 13
0
def test_einsum_inner(a, b):
    return einsum("j,j", a, b)
Esempio n. 14
0
def test_einsum_elemwise1d(a, b):
    return einsum("i,i->i", a, b)
Esempio n. 15
0
def test_einsum_inner2d(a, b):
    return einsum("ij,kj->ik", a, b)
Esempio n. 16
0
def test_einsum_sum2d_0(a):
    return einsum("ij->i", a)
Esempio n. 17
0
def test_einsum_rowmul(a, b):
    return einsum("ij,kj->ikj", a, b)
Esempio n. 18
0
def test_einsum_outer2d(a, b):
    return einsum("ij,kl->ijkl", a, b)