Skip to content

Commit 2a36f0a

Browse files
authored
Merge pull request #13784 from tensor-tang/fix/lodtensor
fix lod tensor
2 parents 6094a72 + dba6928 commit 2a36f0a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/fluid/lod_tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def create_lod_tensor(data, recursive_seq_lens, place):
7474
assert [
7575
new_recursive_seq_lens
7676
] == recursive_seq_lens, "data and recursive_seq_lens do not match"
77-
flattened_data = np.concatenate(data, axis=0).astype("int64")
77+
flattened_data = np.concatenate(data, axis=0)
7878
flattened_data = flattened_data.reshape([len(flattened_data), 1])
7979
return create_lod_tensor(flattened_data, recursive_seq_lens, place)
8080
elif isinstance(data, np.ndarray):

0 commit comments

Comments
 (0)