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")
