Method test_reduceMean_gpu

test/python/test_onnx.py:1722–1723

(self)

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

    def _squeeze_helper(self, dev):
        X = np.random.randn(3, 1, 2, 1, 1)