Beispiel #1
0
def test_full_like_with_device(device):
    t = chainerx.empty((2,), 'float32')
    a = chainerx.full_like(t, 1, device)
    b = chainerx.full_like(t, 1)
    array_utils.check_device(a, device)
    chainerx.testing.assert_array_equal_ex(a, b)
Beispiel #2
0
def test_full_like_with_device(device):
    t = chainerx.empty((2,), 'float32')
    a = chainerx.full_like(t, 1, device)
    b = chainerx.full_like(t, 1)
    array_utils.check_device(a, device)
    chainerx.testing.assert_array_equal_ex(a, b)