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_scalar_cast():
assert isinstance(scalar_cast(1.5, type_to_abstract(i64)), np.int64)
assert isinstance(scalar_cast(1.5, type_to_abstract(f16)), np.float16)
def test_cast_to_dtype(arr, x):
return scalar_cast(x, dtype(arr))