def test_dtype(self): paddle.enable_static() prog = paddle.static.Program() with paddle.static.program_guard(prog): out_1 = paddle.logspace(0, 10, 5, 2, dtype='float32') out_2 = paddle.logspace(0, 10, 5, 2, dtype=np.float32) exe = paddle.static.Executor() res_1, res_2 = exe.run(prog, fetch_list=[out_1, out_2]) assert np.array_equal(res_1, res_2) paddle.disable_static()
def test_imperative(self): paddle.disable_static() out1 = paddle.logspace(0, 10, 5, 2, dtype='float32') np_out1 = np.logspace(0, 10, 5, base=2, dtype='float32') out2 = paddle.logspace(0, 10, 5, 2, dtype='int32') np_out2 = np.logspace(0, 10, 5, base=2, dtype='int32') out3 = paddle.logspace(0, 10, 200, 2, dtype='int32') np_out3 = np.logspace(0, 10, 200, base=2, dtype='int32') paddle.enable_static() self.assertEqual((out1.numpy() == np_out1).all(), True) self.assertEqual((out2.numpy() == np_out2).all(), True) self.assertEqual((out3.numpy() == np_out3).all(), True)
def test_name(self): with paddle.static.program_guard(paddle.static.Program()): out = paddle.logspace(0, 10, 5, 2, dtype='float32', name='logspace_res') assert 'logspace_res' in out.name
def test_variable_input2(self): paddle.disable_static() start = paddle.full(shape=[1], fill_value=0, dtype='float32') stop = paddle.full(shape=[1], fill_value=10, dtype='float32') num = paddle.full(shape=[1], fill_value=5, dtype='int32') base = paddle.full(shape=[1], fill_value=2, dtype='float32') out = paddle.logspace(start, stop, num, base, dtype='float32') np_res = np.logspace(0, 10, 5, base=2, dtype='float32') self.assertEqual((out.numpy() == np_res).all(), True) paddle.enable_static()
def test_variable_input1(self): paddle.enable_static() prog = paddle.static.Program() with paddle.static.program_guard(prog): start = paddle.full(shape=[1], fill_value=0, dtype='float32') stop = paddle.full(shape=[1], fill_value=10, dtype='float32') num = paddle.full(shape=[1], fill_value=5, dtype='int32') base = paddle.full(shape=[1], fill_value=2, dtype='float32') out = paddle.logspace(start, stop, num, base, dtype='float32') exe = paddle.static.Executor() res = exe.run(prog, fetch_list=[out]) np_res = np.logspace(0, 10, 5, base=2, dtype='float32') self.assertEqual((res == np_res).all(), True) paddle.disable_static()
def test_base_dtype(): base = paddle.static.data(shape=[1], dtype="float64", name="end") paddle.logspace(0, 10, 1, base, dtype="float32")
def test_num_dtype(): num = paddle.static.data(shape=[1], dtype="float32", name="step") paddle.logspace(0, 10, num, 2, dtype="float32")
def test_end_dtype(): end = paddle.static.data(shape=[1], dtype="float64", name="end") paddle.logspace(0, end, 1, 2, dtype="float32")
def test_start_dtype(): start = paddle.static.data(shape=[1], dtype="float64", name="start") paddle.logspace(start, 10, 1, 2, dtype="float32")
def test_num_type(): paddle.logspace(0, 10, [0], 2, dtype="float32")
def test_end_type(): paddle.logspace(0, [10], 1, 2, dtype="float32")
def test_start_type(): paddle.logspace([0], 10, 1, 2, dtype="float32")
def test_dtype1(): paddle.logspace(0, 10, 1.33, 2, dtype="int32")
def test_dtype(): paddle.logspace(0, 10, 1, 2, dtype="int8")