Function ParseArrayToTensor

oneflow/api/python/functional/indexing.cpp:128–140

}

void ParseArrayToTensor(PyObject* object,
                        const std::shared_ptr<vm::EagerBlobObject>& eager_blob_object) {
  const DataType dtype = eager_blob_object->data_type();
  const int ndims = eager_blob_object->shape().NumAxes();
  DimVector strides(ndims);
  int64_t size = 1;
  for (int i = ndims - 1; i >= 0; --i) {
    strides[i] = size;
    size *= eager_blob_object->shape().At(i);
  }
  RecursiveParseAndAssign(object, eager_blob_object->mut_dptr<char>(), ndims, 0,
                          eager_blob_object->shape(), strides, dtype);
}

Shape InferArraySizes(PyObject* object) {
  DimVector sizes;