In __getitem__, convert integers to int64 Tensor not int32 to be compatible with Lite(#31658)

4 years ago · 402288ad65
parent 2fbe9b097a
commit 402288ad65
1 changed files with 1 additions and 1 deletions
--- 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