未验证 提交 402288ad 编写于 作者: L liym27 提交者: GitHub

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

上级 2fbe9b09
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册