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