diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index 5ee7b04248e4527357060839a769f7a2c726d744..84d3f5547feb4b5c4fa6d3d9b88a57b9a1e52344 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -1763,7 +1763,10 @@ class Variable(object): Examples: .. code-block:: python + import paddle import paddle.fluid as fluid + + paddle.enable_static() cur_program = fluid.Program() cur_block = cur_program.current_block() new_variable = cur_block.create_var(name="X", @@ -1773,7 +1776,8 @@ class Variable(object): """ if self.type == core.VarDesc.VarType.SELECTED_ROWS: raise Exception("SelectedRows DO NOT supprt lod") - + if self.type == core.VarDesc.VarType.STRINGS: + return None return self.desc.lod_level() @property diff --git a/python/paddle/fluid/tests/unittests/test_variable.py b/python/paddle/fluid/tests/unittests/test_variable.py index 5ba54daa0d4cbc49d4693090a853347f2e4355ab..a3bfe3864a2493fdcf100a1a86648a159701ec11 100644 --- a/python/paddle/fluid/tests/unittests/test_variable.py +++ b/python/paddle/fluid/tests/unittests/test_variable.py @@ -63,6 +63,12 @@ class TestVariable(unittest.TestCase): self.assertRaises(ValueError, lambda: b.create_var(name="fc.w", shape=(24, 100))) + w = b.create_var( + dtype=paddle.fluid.core.VarDesc.VarType.STRINGS, + shape=[1], + name="str_var") + self.assertEqual(None, w.lod_level) + def test_element_size(self): with fluid.program_guard(Program(), Program()): x = paddle.static.data(name='x1', shape=[2], dtype='bool')