def test_transpose2(x, y, axis1, axis2):
perm = (scalar_cast(axis1, u64), scalar_cast(axis2, u64))
xt = transpose(x, perm)
yt = transpose(y, perm)
d = dot(xt, yt)
sm = array_reduce(scalar_add, d, ())
return array_to_scalar(sm)
def test_transpose(v, perm):
return transpose(v, perm)
def test_transpose(x):
return transpose(x, (1, 0))
def before3(x):
return transpose(x, (1, 0))
def before2(x):
x = transpose(x, (1, 0))
x = transpose(x, (1, 0))
return x
def before1(x):
return transpose(x, (0, 1))
def test_prim_transpose():
assert transpose(np.empty((2, 3)), (1, 0)).shape == (3, 2)
assert transpose(np.empty((2, 3, 4)), (2, 0, 1)).shape == (4, 2, 3)
def test_transpose(x, y):
xt = transpose(x, (1, 0))
yt = transpose(y, (1, 0))
d = dot(xt, yt)
sm = array_reduce(scalar_add, d, ())
return array_to_scalar(sm)
