Function TensorSize_AsShape

oneflow/api/python/framework/size.cpp:182–194

}

Shape TensorSize_AsShape(PyObject* self) {
  if (!TensorSize_Check(self)) {
    PyErr_Format(PyExc_TypeError, "can only convert TensorSize(not \"%s\") to Shape",
                 Py_TYPE(self)->tp_name);
    return Shape();
  }
  int size = TensorSize_length((TensorSize*)self);
  DimVector dim_vec(size);
  for (int i = 0; i < size; ++i) {
    dim_vec[i] = PyLong_AsLongLong(PyTuple_GET_ITEM((TensorSize*)self, i));
  }
  return Shape(std::move(dim_vec));
}

ONEFLOW_API_PYBIND11_MODULE("", m) {
  if (PyType_Ready(&TensorSize_Type) < 0) { return; }