Merge pull request #13784 from tensor-tang/fix/lodtensor

fix lod tensor
6 years ago · 2a36f0a3c8
parent 6094a72308 dba6928775
commit 2a36f0a3c8
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/lod_tensor.py
+++ b/python/paddle/fluid/lod_tensor.py
@ -74,7 +74,7 @@ def create_lod_tensor(data, recursive_seq_lens, place):
        assert [
            new_recursive_seq_lens
        ] == recursive_seq_lens, "data and recursive_seq_lens do not match"
-        flattened_data = np.concatenate(data, axis=0).astype("int64")
+        flattened_data = np.concatenate(data, axis=0)
        flattened_data = flattened_data.reshape([len(flattened_data), 1])
        return create_lod_tensor(flattened_data, recursive_seq_lens, place)
    elif isinstance(data, np.ndarray):