def test_gpu_kernel(): paddle.set_device('gpu') tensor = paddle.to_tensor(self.inputs) for axis in self.axises: value_expect, indice_expect = cal_mode(self.inputs, axis) v, inds = paddle.mode(tensor, axis) self.assertTrue(np.allclose(v.numpy(), value_expect)) value_expect, indice_expect = cal_mode( self.inputs, axis, keepdim=True) v, inds = paddle.mode(tensor, axis, keepdim=True) self.assertTrue(np.allclose(v.numpy(), value_expect))
def test_run_static(self): paddle.enable_static() with paddle.static.program_guard(paddle.static.Program(), paddle.static.Program()): input_tensor = paddle.static.data( name="x", shape=[2, 10, 10], dtype="float64") result = paddle.mode(input_tensor, axis=1) expect_value = cal_mode(self.input_data, axis=1)[0] exe = paddle.static.Executor(paddle.CPUPlace()) paddle_result = exe.run(feed={"x": self.input_data}, fetch_list=[result])[0] self.assertTrue(np.allclose(paddle_result, expect_value))