Method _round

test/python/test_api.py:895–899

(self, dev=gpu_dev)

        self._round(gpu_dev)

    def _round(self, dev=gpu_dev):
        x = tensor.Tensor(shape=(3,4,5), device=dev).gaussian(0, 1)
        y = tensor._call_singa_func(singa_api.Round, x.data)
        np.testing.assert_array_almost_equal(np.round(tensor.to_numpy(x)),
                                             tensor.to_numpy(y))

    def test_round_even_cpu(self):
        self._round_even(cpu_dev)