Python astensor Examples

Example #1

File: test_tensor.py Project: Lily977/chainer-pytorch-migration

def test_astensor_empty_gpu():
    a = cupy.array([], dtype=cupy.float32)
    t = tensor.astensor(a)
    assert isinstance(t, torch.Tensor)
    assert t.device.type == 'cuda'
    t += 1
    numpy.testing.assert_array_equal(a.get(), t.cpu().numpy())

Example #2

File: test_tensor.py Project: toslunar/chainer-pytorch-migration

def test_astensor_empty_cpu():
    a = numpy.array([], dtype=numpy.float32)
    t = tensor.astensor(a)

Example #3

File: test_tensor.py Project: toslunar/chainer-pytorch-migration

def test_astensor_negative_stride():
    a = numpy.array([1, 2, 3])
    a = a[::-1]
    t = tensor.astensor(a)
    numpy.testing.assert_array_equal(a, t.numpy())

Example #4

File: test_tensor.py Project: toslunar/chainer-pytorch-migration

def test_astensor_gpu():
    a = cupy.arange(5, dtype=cupy.float32)
    t = tensor.astensor(a)
    assert isinstance(t, torch.Tensor)
    t += 1
    numpy.testing.assert_array_equal(a.get(), t.cpu().numpy())

Example #5

File: test_tensor.py Project: Lily977/chainer-pytorch-migration

def test_astensor_empty_cpu():
    a = numpy.array([], dtype=numpy.float32)
    t = tensor.astensor(a)
    assert t.device.type == 'cpu'