Python linspace примеры использования

Пример #1

Файл: test_creation.py Проект: jnishi/chainer

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

Файл: test_creation.py Проект: dselivanov/chainer

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)