From 6abe7dcb0d40069b46eff6a27b847c3daee7080e Mon Sep 17 00:00:00 2001 From: zyfncg Date: Wed, 1 Dec 2021 21:57:38 +0800 Subject: [PATCH] support bool index for int tensor (#37761) --- python/paddle/fluid/variable_index.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/python/paddle/fluid/variable_index.py b/python/paddle/fluid/variable_index.py index 19067b8ae1..5aa7f9c972 100644 --- a/python/paddle/fluid/variable_index.py +++ b/python/paddle/fluid/variable_index.py @@ -674,14 +674,6 @@ def _setitem_impl_(var, item, value): # the item is a tensor of bool def set_value_for_bool_tensor(var, item, value): - - # TODO(zyfncg): Now scatter_nd_add only support float32 and float64 tensor, - # so in the current version we also only support float32 and float64 tensor, - # this problem will be fixed in the future. - if var.dtype != core.VarDesc.VarType.FP32 and var.dtype != core.VarDesc.VarType.FP64: - raise TypeError("Only support float and double tensor for bool index, " - "but received {}.".format(var.dtype)) - if len(item.shape) > len(var.shape): raise IndexError("The dims of bool index doesn't match indexed array, " "the dims of bool index except to be equal or less " -- GitLab