Python empty_like示例

示例#1

文件： test_creation.py 项目： jnishi/chainer

def test_empty_like_with_device(device):
    t = chainerx.empty((2,), 'float32')
    a = chainerx.empty_like(t, device)
    b = chainerx.empty_like(t)
    array_utils.check_device(a, device)
    assert a.dtype == b.dtype
    assert a.shape == b.shape

示例#2

文件： test_creation.py 项目： dselivanov/chainer

def test_empty_like_with_device(device):
    t = chainerx.empty((2,), 'float32')
    a = chainerx.empty_like(t, device)
    b = chainerx.empty_like(t)
    array_utils.check_device(a, device)
    assert a.dtype == b.dtype
    assert a.shape == b.shape