Python identity示例

示例#1

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

def test_identity_with_device(device):
    a = chainerx.identity(3, 'float32', device)
    b = chainerx.identity(3, 'float32')
    array_utils.check_device(a, device)
    chainerx.testing.assert_array_equal_ex(a, b)

示例#2

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

def test_identity_with_device(device):
    a = chainerx.identity(3, 'float32', device)
    b = chainerx.identity(3, 'float32')
    array_utils.check_device(a, device)
    chainerx.testing.assert_array_equal_ex(a, b)