Method test_mean_gpu

test/python/test_onnx.py:1187–1188

(self)

    @unittest.skipIf(not singa_api.USE_CUDA, 'CUDA is not enabled')
    def test_mean_gpu(self):
        self._mean_helper(gpu_dev)

    def _pow_helper(self, dev):
        x0 = np.array([7, 5, 0.2, 0.1, 0.3, 4]).reshape(3, 2).astype(np.float32)