Python Variable.copy_fromの例

コード例 #1

def test_copy_from_cpu():
    src = Variable(rand((100, )))

    dest = Variable(rand((100, )))
    dest.copy_from(src)

    close(src, dest)

コード例 #2

def test_copy_from_gpu():
    set_cuda_active(True)

    src = Variable(rand((100, )))
    src.to_gpu()

    dest = Variable(rand((100, )))
    dest.to_gpu()

    dest.copy_from(src)
    close(src, dest)

    close(src._gpu.new_array(), dest._gpu.new_array())

コード例 #3

def test_copy_from_another_gpu():
    set_cuda_active(True)

    src = Variable(rand((100, )))
    src.to_gpu()

    with use_device(1):
        dest = Variable(rand((100, )))
        dest.to_gpu()

    dest.copy_from(src)
    close(src, dest)

    close(src._gpu.new_array(), dest._gpu.new_array())