示例#1
0
def test_scalar_to_array():
    a = scalar_to_array(1)
    assert isinstance(a, np.ndarray)
    assert a.dtype == np.int64
    b = scalar_to_array(1.5)
    assert isinstance(b, np.ndarray)
    assert b.dtype == np.float64
示例#2
0
def test_scalar_to_array(x):
    return scalar_to_array(x)
示例#3
0
def test_distribute2(x):
    return distribute(scalar_to_array(x), (1, ))
示例#4
0
def test_distribute(x):
    return distribute(scalar_to_array(x), (2, 3))
示例#5
0
 def up1(x2):
     return array_map(
         scalar_add, x2,
         distribute(scalar_to_array(1, typeof(x)), shape(x)))
示例#6
0
 def after(xs):
     return distribute(scalar_to_array(3, arr_t), shape(xs))
示例#7
0
 def up1(x):
     return array_map(scalar_add, x,
                      distribute(scalar_to_array(1), (2, 3)))
示例#8
0
 def after(xs):
     return distribute(scalar_to_array(3), (3, 5))
示例#9
0
def test_array_operations_distribute(x, y):
    xs = distribute(scalar_to_array(x), (4, 3))
    ys = distribute(scalar_to_array(y), (4, 3))
    div = array_map(scalar_div, xs, ys)
    sm = array_reduce(scalar_add, div, ())
    return array_to_scalar(sm)