diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index 04ed384846fb66da935ba9782a518b3c0707fa7b..036e8ab30441cb3bf3478871b6368f93835d23aa 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -877,7 +877,7 @@ def _getitem_impl_(var, item): new_list_tensor.append(dim) else: assert (isinstance(dim, int)) - temp_out = var.block.create_var(dtype='int32') + temp_out = var.block.create_var(dtype='int64') fill_constant([1], dim, force_cpu=True, out=temp_out) new_list_tensor.append(temp_out) return new_list_tensor