예제 #1
0
def test_linspace_with_device(device):
    a = chainerx.linspace(3, 5, 10, dtype='float32', device=device)
    b = chainerx.linspace(3, 5, 10, dtype='float32')
    array_utils.check_device(a, device)
    chainerx.testing.assert_array_equal_ex(a, b)
예제 #2
0
def test_linspace_with_device(device):
    a = chainerx.linspace(3, 5, 10, dtype='float32', device=device)
    b = chainerx.linspace(3, 5, 10, dtype='float32')
    array_utils.check_device(a, device)
    chainerx.testing.assert_array_equal_ex(a, b)