fix ut

Author: zoooo0820

paddle/fluid/pybind/slice_utils.h

@@ -609,9 +609,9 @@ static paddle::Tensor dealWithValues(const paddle::Tensor& tensor,
  tensor.dtype() == phi::DataType::INT16 ||
  tensor.dtype() == phi::DataType::INT8 ||
  tensor.dtype() == phi::DataType::UINT8) {
- values->push_back(value_obj_tmp.cast<int32_t>());
+ values->push_back(value_obj_tmp.cast<float>());
  } else if (tensor.dtype() == phi::DataType::INT64) {
- values->push_back(value_obj_tmp.cast<int64_t>());
+ values->push_back(value_obj_tmp.cast<double>());
  } else if (tensor.dtype() == phi::DataType::BOOL) {
  values->push_back(value_obj_tmp.cast<bool>());
  } else if (tensor.dtype() == phi::DataType::COMPLEX64) {

python/paddle/base/dygraph/tensor_patch_methods.py

@@ -975,7 +975,7 @@ def __array__(self, dtype=None):
  array = array.astype(dtype)
  return array
 
- def pre_deal_index(self, item, value=None):
+ def pre_deal_index(self, item):
  # since in pybind there is no effiency way to transfer Py_Tuple/Py_List/Py_Range to Tensor
  # we call this function in python level.
  item = list(item) if isinstance(item, tuple) else [item]
@@ -985,14 +985,14 @@ def pre_deal_index(self, item, value=None):
  elif isinstance(slice_item, range):
  item[i] = paddle.to_tensor(list(slice_item))
 
- return tuple(item), value
+ return tuple(item)
 
  def __getitem__(self, item):
- item, _ = pre_deal_index(self, item)
+ item = pre_deal_index(self, item)
  return self._getitem_dygraph(item)
 
  def __setitem__(self, item, value):
- item, value = pre_deal_index(self, item, value)
+ item = pre_deal_index(self, item)
  return self._setitem_dygraph(item, value)
 
  @framework.dygraph_only