Exemplos de ndarray em Python

Exemplo n.º 1

Arquivo: test_cuda.py Projeto: y1r/chainer

    def test_from_array(self, backend_config):
        with cuda.Device(backend_config.cuda_device):
            arr = cuda.ndarray((), numpy.float32)
        # Test precondition check
        assert arr.device.id == backend_config.cuda_device

        device = backend.GpuDevice.from_array(arr)
        self.check_device(device, backend_config)
        assert device == backend.GpuDevice.from_device_id(
            backend_config.cuda_device)

Exemplo n.º 2

Arquivo: test_cuda.py Projeto: asi1024/chainer

    def test_from_array(self, backend_config):
        with cuda.Device(backend_config.cuda_device):
            arr = cuda.ndarray((), numpy.float32)
        # Test precondition check
        assert arr.device.id == backend_config.cuda_device

        device = backend.GpuDevice.from_array(arr)
        assert isinstance(device, backend.GpuDevice)
        assert device == backend.GpuDevice.from_device_id(
            backend_config.cuda_device)

Exemplo n.º 3

Arquivo: test_cuda.py Projeto: PaulGureghian1/Chainer

    def test_from_array(self, backend_config):
        with cuda.Device(backend_config.cuda_device):
            arr = cuda.ndarray((), numpy.float32)
        # Test precondition check
        assert arr.device.id == backend_config.cuda_device

        device = backend.GpuDevice.from_array(arr)
        assert isinstance(device, backend.GpuDevice)
        assert (device
                == chainer.get_device((cuda.cupy, backend_config.cuda_device)))

Exemplo n.º 4

    def test_get_device_from_array(self, backend_config):
        with cuda.Device(backend_config.cuda_device):
            arr = cuda.ndarray((), numpy.float32)
        # Test precondition check
        assert arr.device.id == backend_config.cuda_device

        expected_device = backend_config.device

        device = backend.GpuDevice.from_array(arr)
        assert device == expected_device

        device = backend.get_device_from_array(arr)
        assert device == expected_device

Exemplo n.º 5

Arquivo: test_cuda.py Projeto: asi1024/chainer

    def test_get_device_from_array(self, backend_config):
        with cuda.Device(backend_config.cuda_device):
            arr = cuda.ndarray((), numpy.float32)
        # Test precondition check
        assert arr.device.id == backend_config.cuda_device

        expected_device = backend_config.device

        device = backend.GpuDevice.from_array(arr)
        assert device == expected_device

        device = backend.get_device_from_array(arr)
        assert device == expected_device

Exemplo n.º 6

 def test_array_gpu(self):
     self._check_array(cuda.ndarray([1, 2]), 'constant array')

Exemplo n.º 7

Arquivo: test_basic_math.py Projeto: unnonouno/chainer

 def test_array_gpu(self):
     self._check_array(cuda.ndarray([1, 2]), 'constant array')